put mast machine mapping import into try-except

gafusion · Jun 5, 2024 · a4a9bff · a4a9bff
1 parent c232397
commit a4a9bff
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/omas/omas_machine.py b/omas/omas_machine.py
@@ -6,7 +6,7 @@
 from .omas_utils import *
 from .omas_core import ODS, dynamic_ODS, omas_environment, omas_info_node, imas_json_dir, omas_rcparams
 from .omas_physics import cocos_signals
-from omas.machine_mappings import d3d, nstx, nstxu, east, mast
+from omas.machine_mappings import d3d, nstx, nstxu, east
 from omas.machine_mappings.d3d import __regression_arguments__
 from omas.utilities.machine_mapping_decorator import machine_mapping_function
 from omas.utilities.omas_mds import mdsvalue
@@ -16,6 +16,12 @@
 except:
     pass
 
+try:
+    from omas.machine_mappings import mast
+except ImportError:
+    pass
+
+
 __all__ = [
     'machine_expression_types',
     'machines',