fall back to getattr

openforcefield · Nov 7, 2024 · ce84b5e · ce84b5e
1 parent 4ea6d3a
commit ce84b5e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/openff/toolkit/__init__.py b/openff/toolkit/__init__.py
@@ -79,7 +79,10 @@ def __getattr__(name):
     obj_mod = _lazy_imports_obj.get(name)
     if obj_mod is not None:
         mod = importlib.import_module(obj_mod)
-        return mod.__dict__[name]
+        try:
+            return mod.__dict__[name]
+        except KeyError:  # account for lazy loaders
+            return getattr(mod, name)
 
     lazy_mod = _lazy_imports_mod.get(name)
     if lazy_mod is not None: