Skip to content

Commit 838d543

Browse files
Addressing review comments
1 parent eaf4822 commit 838d543

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

ads/aqua/extension/model_handler.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,12 +164,13 @@ def put(self, id):
164164

165165
enable_finetuning = input_data.get("enable_finetuning")
166166
task = input_data.get("task")
167+
app=AquaModelApp()
167168
self.finish(
168-
AquaModelApp().edit_registered_model(
169+
app.edit_registered_model(
169170
id, inference_container, enable_finetuning, task
170171
)
171172
)
172-
AquaModelApp().clear_model_details_cache(model_id=id)
173+
app.clear_model_details_cache(model_id=id)
173174

174175

175176
class AquaModelLicenseHandler(AquaAPIhandler):

ads/aqua/model/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -726,7 +726,7 @@ def clear_model_details_cache(self, model_id):
726726
dict with the key used, and True if cache has the key that needs to be deleted.
727727
"""
728728
res = {}
729-
logger.info("Clearing _service_model_details_cache")
729+
logger.info(f"Clearing _service_model_details_cache for {model_id}")
730730
with self._cache_lock:
731731
if model_id in self._service_model_details_cache:
732732
self._service_model_details_cache.pop(key=model_id)

0 commit comments

Comments
 (0)