File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -752,12 +752,12 @@ def _fetch_class_library_tuple(module):
752752def _identify_model_variants (folder : str , variant : str , config : dict ) -> dict :
753753 model_variants = {}
754754 if variant is not None :
755- for folder in os .listdir (folder ):
756- folder_path = os .path .join (folder , folder )
757- is_folder = os .path .isdir (folder_path ) and folder in config
755+ for sub_folder in os .listdir (folder ):
756+ folder_path = os .path .join (folder , sub_folder )
757+ is_folder = os .path .isdir (folder_path ) and sub_folder in config
758758 variant_exists = is_folder and any (p .split ("." )[1 ].startswith (variant ) for p in os .listdir (folder_path ))
759759 if variant_exists :
760- model_variants [folder ] = variant
760+ model_variants [sub_folder ] = variant
761761 return model_variants
762762
763763
You can’t perform that action at this time.
0 commit comments