linear fix

uclid-org · Oct 13, 2020 · 918ad9b · 918ad9b
1 parent a6a538a
commit 918ad9b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/medleysolver/classifiers.py b/medleysolver/classifiers.py
@@ -138,8 +138,9 @@ def get_ordering(self, point, count):
                 for i in range(len(SOLVERS))]
 
         ps = [thetas[i].T @ point + beta.T @ point + self.alpha * np.sqrt(sigmas[i]) for i in range(len(SOLVERS))]
-
-        i_order = sorted(random.shuffle(list(range(len(ps)))), key=lambda x: -1 * ps[x])
+        ss = list(range(len(ps)))
+        random.shuffle(ss)
+        i_order = sorted(ss, key=lambda x: -1 * ps[x])
         order = [list(SOLVERS.keys())[int(choice)] for choice in i_order]
         return order