minimise diff

MarcoGorelli · MarcoGorelli · commit 5478a73ad5a9 · 2020-12-09T15:05:45.000Z
diff --git a/pymc3/sampling.py b/pymc3/sampling.py
@@ -1527,7 +1527,7 @@ def _choose_chains(traces, tune):
         raise ValueError("Not enough samples to build a trace.")
 
     idxs = np.argsort(lengths)
-    l_sort = lengths[idxs]
+    l_sort = np.array(lengths)[idxs]
 
     use_until = np.argmax(l_sort * np.arange(1, l_sort.shape[0] + 1)[::-1])
     final_length = l_sort[use_until]
