Merge pull request #286 from pyGSTio/bugfix-germsel-initialization

Bugfix germsel initialization
sandialabs · Dec 21, 2022 · 58a51e0 · 58a51e0
2 parents 5e09748 + 7ea6216
commit 58a51e0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pygsti/algorithms/germselection.py b/pygsti/algorithms/germselection.py
@@ -1479,8 +1479,7 @@ def find_germs_depthfirst(model_list, germs_list, randomize=True,
     if force:
         if force == "singletons":
             weights[_np.where(germLengths == 1)] = 1
-            goodGerms = [germ for germ
-                         in _np.array(germs_list)[_np.where(germLengths == 1)]]
+            goodGerms = [germ for i, germ in enumerate(germs_list) if germLengths[i] == 1]
         else:  # force should be a list of Circuits
             for opstr in force:
                 weights[germs_list.index(opstr)] = 1