From 76b0444c2def109d4a67ce229ab75aadedd2cee3 Mon Sep 17 00:00:00 2001 From: Ralf Weber Date: Tue, 15 Sep 2026 10:12:37 +0100 Subject: [PATCH] Make ray imports in nn_utils optional --- src/ms_pred/nn_utils/__init__.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) 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 *