Use nan mask only once.

HEXRD · Jan 4, 2024 · 3c31131 · 3c31131
1 parent 3331899
commit 3c31131
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hexrd/matrixutil.py b/hexrd/matrixutil.py
@@ -690,8 +690,9 @@ def findDuplicateVectors_numba(vec, tol=vTol, equivPM=False):
     idx = eqv[~np.isnan(eqv)].astype(np.int64)
     uid2 = list(np.delete(uid, idx))
     eqv2 = []
+    mask = ~np.isnan(eqv)
     for ii in range(eqv.shape[1]):
-        v = eqv[ii, ~np.isnan(eqv[ii, :])]
+        v = eqv[ii, mask[ii, :]]
         if v.shape[0] > 0:
             eqv2.append([ii] + list(v.astype(np.int64)))
     return eqv2, uid2