Skip to content

Commit e9a7085

Browse files
authored
Fix fetching model version by UUID passed as string (#3777)
1 parent 23369d9 commit e9a7085

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/zenml/client.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6420,9 +6420,15 @@ def get_model_version(
64206420
if model_version_name_or_number_or_id is None:
64216421
model_version_name_or_number_or_id = ModelStages.LATEST
64226422

6423-
if isinstance(model_version_name_or_number_or_id, UUID):
6423+
if is_valid_uuid(model_version_name_or_number_or_id):
6424+
assert not isinstance(model_version_name_or_number_or_id, int)
6425+
model_version_id = (
6426+
UUID(model_version_name_or_number_or_id)
6427+
if isinstance(model_version_name_or_number_or_id, str)
6428+
else model_version_name_or_number_or_id
6429+
)
64246430
return self.zen_store.get_model_version(
6425-
model_version_id=model_version_name_or_number_or_id,
6431+
model_version_id=model_version_id,
64266432
hydrate=hydrate,
64276433
)
64286434
elif isinstance(model_version_name_or_number_or_id, int):

0 commit comments

Comments
 (0)