diff --git a/src/ms_pred/nn_utils/__init__.py b/src/ms_pred/nn_utils/__init__.py
index dc3a880e..fef9a3f6 100644
--- a/src/ms_pred/nn_utils/__init__.py
+++ b/src/ms_pred/nn_utils/__init__.py
@@ -1,7 +1,15 @@
 from .nn_utils import *
-from .tune_utils import *
+try:
+    from .tune_utils import *
+except ModuleNotFoundError as err:
+    if err.name != "ray":
+        raise
 from .mol_graph import *
-from .base_hyperopt import *
+try:
+    from .base_hyperopt import *
+except ModuleNotFoundError as err:
+    if err.name != "ray":
+        raise
 from .transformer_layer import *
 from .form_embedder import *
 from .dgl_graph_ops import *