job and model rest schema update (#459)

Signed-off-by: Igor Davidyuk <[email protected]>
openvinotoolkit · Jul 4, 2024 · 9cb0aac · 9cb0aac
1 parent 01832f5
commit 9cb0aac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/geti_sdk/data_models/job.py b/geti_sdk/data_models/job.py
@@ -207,6 +207,7 @@ class JobCancellationInfo:
     :var cancel_time: Time at which the Job was cancelled
     """
 
+    cancellable: bool = True
     is_cancelled: bool = False
     user_uid: Optional[str] = None
     cancel_time: Optional[str] = attr.field(converter=str_to_datetime, default=None)

diff --git a/geti_sdk/data_models/model.py b/geti_sdk/data_models/model.py
@@ -229,8 +229,11 @@ class Model(BaseModel):
 
     architecture: str = attr.field(kw_only=True)
     score_up_to_date: bool = attr.field(kw_only=True)
-    optimization_capabilities: OptimizationCapabilities = attr.field(kw_only=True)
     optimized_models: List[OptimizedModel] = attr.field(kw_only=True)
+    # Removed in Geti 2.2
+    optimization_capabilities: Optional[OptimizationCapabilities] = attr.field(
+        default=None, kw_only=True
+    )
     labels: Optional[List[Label]] = None
     version: Optional[int] = attr.field(default=None, kw_only=True)
     # 'version' is deprecated in v1.1 -- IS IT?