Don't except on load_class import

grc-iit · Oct 2, 2023 · 30b8a54 · 30b8a54
1 parent 6fbba07
commit 30b8a54
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/jarvis_util/util/import_mod.py b/jarvis_util/util/import_mod.py
@@ -24,7 +24,10 @@ def load_class(import_str, path, class_name):
  """
  if len(path):
  sys.path.insert(0, path)
- module = __import__(import_str, fromlist=[class_name])
+ try:
+ module = __import__(import_str, fromlist=[class_name])
+ except:
+ return None
  if len(path):
  sys.path.pop(0)
  return getattr(module, class_name)