nblast_smart: make mask initialization use less memory

navis-org · Mar 28, 2022 · d656cb7 · d656cb7
1 parent b135f52
commit d656cb7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/navis/nbl/nblast_funcs.py b/navis/nbl/nblast_funcs.py
@@ -486,7 +486,7 @@ def nblast_smart(query: Union[Dotprops, NeuronList],
         # Sort such that the top hit is to the left
         srt = np.argsort(scr.values, axis=1)[:, ::-1]
         # Generate the mask
-        mask = pd.DataFrame(np.zeros(scr.shape), dtype=bool,
+        mask = pd.DataFrame(np.zeros(scr.shape, dtype=bool),
                             columns=scr.columns, index=scr.index)
         _ = np.arange(mask.shape[0])
         for N in range(t):