We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d458660 commit bc519e3Copy full SHA for bc519e3
1 file changed
hwcomponents/find_models.py
@@ -239,7 +239,7 @@ def get_models(
239
models.extend(new_models)
240
241
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)
+ models = [m for m in models if name_must_include.lower() in m.model_name.lower()]
+ return sorted(models, key=lambda x: x.model_name)
244
models = [m for m in models if name_must_include.lower() in m.__name__.lower()]
245
return sorted(models, key=lambda x: x.__name__)
0 commit comments