Skip to content

Commit d1e9608

Browse files
Updating edit registered model flow
1 parent 1b5bbd1 commit d1e9608

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ads/aqua/model/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -828,8 +828,8 @@ def clear_model_details_cache(self, model_id):
828828

829829
@staticmethod
830830
def list_valid_inference_containers():
831-
containers = list(AquaApp.get_container_config())
832-
family_values = [item.family_name for item in containers]
831+
containers = AquaApp().get_container_config().to_dict().get("inference")
832+
family_values = [item.family for item in containers]
833833
return family_values
834834

835835
@telemetry(

0 commit comments

Comments
 (0)