feat(client): use AttrMixin to handle the default values for Job attrs

PR Closed: #1182
Graviti-AI · Dec 24, 2021 · 7f4a2a5 · 7f4a2a5
1 parent 846bf57
commit 7f4a2a5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/tensorbay/client/job.py b/tensorbay/client/job.py
@@ -55,11 +55,11 @@ class Job(AttrsMixin, ReprMixin):  # pylint: disable=too-many-instance-attribute
     job_id: str = attr(key=camel)
     arguments: Dict[str, Any] = attr()
     created_at: int = attr(key=camel)
-    started_at: Optional[int] = attr(key=camel)
-    finished_at: Optional[int] = attr(key=camel)
+    started_at: Optional[int] = attr(key=camel, default=None)
+    finished_at: Optional[int] = attr(key=camel, default=None)
     status: str = attr()
-    error_message: str = attr(key=camel)
-    _result: Optional[Dict[str, Any]] = attr(key="result")
+    error_message: str = attr(key=camel, default="")
+    _result: Optional[Dict[str, Any]] = attr(key="result", default=None)
     description: Optional[str] = attr(default="")
 
     def __init__(  # pylint: disable=too-many-arguments

diff --git a/tensorbay/client/version.py b/tensorbay/client/version.py
@@ -554,12 +554,8 @@ def _create_job(
         response.update(
             title=title,
             arguments=arguments,
-            startedAt=None,
-            finishedAt=None,
             status="QUEUING",
-            errorMessage="",
             description=description,
-            result=None,
         )
         return response