Skip to content

Commit f54afc3

Browse files
committed
Add default args to Model.save
1 parent 2ce5812 commit f54afc3

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

model_api/python/model_api/adapters/onnx_adapter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def update_model_info(self, model_info: dict[str, Any]):
202202
else:
203203
meta.value = str(model_info[item])
204204

205-
def save_model(self, path: str, weights_path: str | None, version: str | None):
205+
def save_model(self, path: str, weights_path: str | None = None, version: str | None = None):
206206
"""
207207
Serializes model to the filesystem.
208208

model_api/python/model_api/adapters/openvino_adapter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -441,7 +441,7 @@ def update_model_info(self, model_info: dict[str, Any]):
441441
for name in model_info:
442442
self.model.set_rt_info(model_info[name], ["model_info", name])
443443

444-
def save_model(self, path: str, weights_path: str | None, version: str | None):
444+
def save_model(self, path: str, weights_path: str | None = None, version: str | None = None):
445445
"""
446446
Saves OV model as two files: .xml (architecture) and .bin (weights).
447447

model_api/python/model_api/adapters/ovms_adapter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def update_model_info(self, model_info: dict[str, Any]):
129129
msg = "OVMSAdapter does not support updating model info"
130130
raise NotImplementedError(msg)
131131

132-
def save_model(self, path: str, weights_path: str | None, version: str | None):
132+
def save_model(self, path: str, weights_path: str | None = None, version: str | None = None):
133133
msg = "OVMSAdapter does not support saving a model"
134134
raise NotImplementedError(msg)
135135

model_api/python/model_api/models/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -548,7 +548,7 @@ def log_layers_info(self):
548548
f"precision: {metadata.precision}, layout: {metadata.layout}",
549549
)
550550

551-
def save(self, path: str, weights_path: str | None, version: str | None):
551+
def save(self, path: str, weights_path: str | None = None, version: str | None = None):
552552
"""
553553
Serializes model to the filesystem. Model format depends in the InferenceAdapter being used.
554554

0 commit comments

Comments
 (0)