Fix numba typing error

Signed-off by: David Rowenhorst <[email protected]>
drowenhorst-nrl · Dec 5, 2024 · f812300 · f812300
1 parent eb8055d
commit f812300
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pyebsdindex/nlpar_cpu.py b/pyebsdindex/nlpar_cpu.py
@@ -326,6 +326,9 @@ def calcnlpar(self, chunksize=0, searchradius=None, lam = None, dthresh = None,
     sr = np.int64(self.searchradius)
     diff_offset = np.float32(self.diff_offset)
 
+    if type(diff_offset) is np.ndarray:
+      diff_offset = np.float32(diff_offset[0])
+
     if filename is not None:
       self.setfile(filepath=filename)