Set constraints to empty dict when it is None

To prevent iterating over None.
SLAC-ML · Nov 2, 2023 · 9fdd2bf · 9fdd2bf
1 parent bd75ec5
commit 9fdd2bf
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/badger/gui/default/components/routine_runner.py b/src/badger/gui/default/components/routine_runner.py
@@ -166,8 +166,11 @@ def get_routine_xopt(self):
         objectives = {key: value for dictionary in routine['config']['objectives']
                       for key, value in dictionary.items()}
         # TODO: consider critical ones
-        constraints = {key: value[:2] for dictionary in routine['config']['constraints']
-                       for key, value in dictionary.items()}
+        if routine['config']['constraints'] is not None:
+            constraints = {key: value[:2] for dictionary in routine['config']['constraints']
+                           for key, value in dictionary.items()}
+        else:
+            constraints = {}
         states = routine['config']['states'] or []
         vocs = {
             'variables': variables,