File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments