Skip to content

Commit 9cb0aac

Browse files
job and model rest schema update (#459)
Signed-off-by: Igor Davidyuk <igor.davidyuk@intel.com>
1 parent 01832f5 commit 9cb0aac

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

geti_sdk/data_models/job.py

+1
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,7 @@ class JobCancellationInfo:
207207
:var cancel_time: Time at which the Job was cancelled
208208
"""
209209

210+
cancellable: bool = True
210211
is_cancelled: bool = False
211212
user_uid: Optional[str] = None
212213
cancel_time: Optional[str] = attr.field(converter=str_to_datetime, default=None)

geti_sdk/data_models/model.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -229,8 +229,11 @@ class Model(BaseModel):
229229

230230
architecture: str = attr.field(kw_only=True)
231231
score_up_to_date: bool = attr.field(kw_only=True)
232-
optimization_capabilities: OptimizationCapabilities = attr.field(kw_only=True)
233232
optimized_models: List[OptimizedModel] = attr.field(kw_only=True)
233+
# Removed in Geti 2.2
234+
optimization_capabilities: Optional[OptimizationCapabilities] = attr.field(
235+
default=None, kw_only=True
236+
)
234237
labels: Optional[List[Label]] = None
235238
version: Optional[int] = attr.field(default=None, kw_only=True)
236239
# 'version' is deprecated in v1.1 -- IS IT?

0 commit comments

Comments
 (0)