Unverified Commit dc07fc29 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix _identify_model_variants (#9247)

merge now, will add/fix test next 
parent 805bf33f
......@@ -752,12 +752,12 @@ def _fetch_class_library_tuple(module):
def _identify_model_variants(folder: str, variant: str, config: dict) -> dict:
model_variants = {}
if variant is not None:
for folder in os.listdir(folder):
folder_path = os.path.join(folder, folder)
is_folder = os.path.isdir(folder_path) and folder in config
for sub_folder in os.listdir(folder):
folder_path = os.path.join(folder, sub_folder)
is_folder = os.path.isdir(folder_path) and sub_folder in config
variant_exists = is_folder and any(p.split(".")[1].startswith(variant) for p in os.listdir(folder_path))
if variant_exists:
model_variants[folder] = variant
model_variants[sub_folder] = variant
return model_variants
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment