Skip to content

Commit bc519e3

Browse files
Fix name error
1 parent d458660 commit bc519e3

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

hwcomponents/find_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ def get_models(
239239
models.extend(new_models)
240240

241241
if _return_wrappers:
242-
models = [m for m in models if name_must_include.lower() in m.name.lower()]
243-
return sorted(models, key=lambda x: x.name)
242+
models = [m for m in models if name_must_include.lower() in m.model_name.lower()]
243+
return sorted(models, key=lambda x: x.model_name)
244244
models = [m for m in models if name_must_include.lower() in m.__name__.lower()]
245245
return sorted(models, key=lambda x: x.__name__)

0 commit comments

Comments
 (0)