try adding ddmd imports within try block

Libensemble · Jul 14, 2021 · 294316e · 294316e
1 parent 387f7d7
commit 294316e
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/libensemble/tests/scaling_tests/ddmd/run_libe_ddmd.py b/libensemble/tests/scaling_tests/ddmd/run_libe_ddmd.py
@@ -14,11 +14,14 @@
 from keras_cvae_ml_genf import run_keras_cvae_ml_genf
 
 # Import DeepDriveMD components for registering with libEnsemble's executor
-from deepdrivemd.sim.openmm import run_openmm
-from deepdrivemd.aggregation.basic import aggregate
-from deepdrivemd.models.keras_cvae import train
-from deepdrivemd.selection.latest import select_model
-from deepdrivemd.agents.lof import lof
+try:
+    from deepdrivemd.sim.openmm import run_openmm
+    from deepdrivemd.aggregation.basic import aggregate
+    from deepdrivemd.models.keras_cvae import train
+    from deepdrivemd.selection.latest import select_model
+    from deepdrivemd.agents.lof import lof
+except ModuleNotFoundError:
+    pass
 
 logger.set_level('INFO')