put back import hack

nomad-coe · Dec 13, 2024 · a14cc38 · a14cc38
1 parent bc06fd1
commit a14cc38
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/simulationworkflowschema/__init__.py b/simulationworkflowschema/__init__.py
@@ -94,7 +94,13 @@ def load_modules():
         'photon_polarization',
         'thermodynamics',
     ]
+    import simulationworkflowschema
 
+    for name in sub_modules:
+        sub_module = importlib.import_module(f'simulationworkflowschema.{name}')
+        for method in sub_module.__dict__:
+            if method in __all__:
+                setattr(simulationworkflowschema, method, sub_module.__dict__[method])
 
 class SimulationWorkflowSchemaEntryPoint(SchemaPackageEntryPoint):
     def load(self):