review suggestion

Co-authored-by: Rahul Tuli <rahul@neuralmagic.com>
neuralmagic · Oct 5, 2023 · 7b3e3bc · 7b3e3bc
1 parent 881e868
commit 7b3e3bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparsezoo/utils/registry.py b/src/sparsezoo/utils/registry.py
@@ -225,7 +225,7 @@ def _import_and_get_value_from_module(module_path: str, value_name: str) -> Any:
     spec.loader.exec_module(module)
 
     # get value from module
-    value = getattr(module, value_name)
+    value = getattr(module, value_name, None)
 
     if not value:
         raise RuntimeError(