Allow errors (with warning) when asking for keys() on machine mappings

gafusion · Mar 13, 2024 · 4ab29a3 · 4ab29a3
1 parent 110ba8b
commit 4ab29a3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/omas/machine_mappings/d3d.json b/omas/machine_mappings/d3d.json
@@ -427,6 +427,7 @@
   "PYTHON": "magnetics_hardware(ods, {pulse})"
  },
  "magnetics.flux_loop.:.flux.data": {
+  "COCOSIO": 11,
   "PYTHON": "magnetics_floops_data(ods, {pulse}, {nref!r})"
  },
  "magnetics.flux_loop.:.flux.data_error_upper": {

diff --git a/omas/omas_machine.py b/omas/omas_machine.py
@@ -998,7 +998,11 @@ def __contains__(self, location):
     def keys(self, location):
         ulocation = (o2u(location) + ".").lstrip('.')
         if ulocation + ':' in machine_mappings(self.kw['machine'], self.kw['branch'], self.kw['user_machine_mappings']):
-            return list(range(self[ulocation + ':']))
+            try:
+                return list(range(self[ulocation + ':']))
+            except Exception as _excp:
+                printe(f'{ulocation}: issue:' + repr(_excp))
+                return []
         else:
             tmp = numpy.unique(
                 [