Fix dynamic machine keys

close #230 mention @bechtt
gafusion · Feb 13, 2023 · 6fb175f · 6fb175f
1 parent 2035fc6
commit 6fb175f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/omas/omas_machine.py b/omas/omas_machine.py
@@ -916,9 +916,9 @@ def __contains__(self, location):
         return ulocation in machine_mappings(self.kw['machine'], self.kw['branch'], self.kw['user_machine_mappings'])
 
     def keys(self, location):
-        ulocation = o2u(location)
-        if ulocation + '.:' in machine_mappings(self.kw['machine'], self.kw['branch'], self.kw['user_machine_mappings']):
-            return list(range(self[ulocation + '.:']))
+        ulocation = o2u(location) + "."
+        if ulocation + ':' in machine_mappings(self.kw['machine'], self.kw['branch'], self.kw['user_machine_mappings']):
+            return list(range(self[ulocation + ':']))
         else:
             tmp = numpy.unique(
                 [