Ensure integer dtypes are returned from log()

mhostetter · Jun 22, 2024 · 0ae8b87 · 0ae8b87
1 parent b2e7155
commit 0ae8b87
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/galois/_domains/_ufunc.py b/src/galois/_domains/_ufunc.py
@@ -516,6 +516,8 @@ def __call__(self, ufunc, method, inputs, kwargs, meta):
         inputs = list(inputs) + [int(self.field.primitive_element)]
         inputs, kwargs = self._view_inputs_as_ndarray(inputs, kwargs)
         output = getattr(self.ufunc, method)(*inputs, **kwargs)
+        if output.dtype == np.object_:
+            output = output.astype(int)
         return output
 
 

diff --git a/src/galois/_fields/_array.py b/src/galois/_fields/_array.py
@@ -1730,6 +1730,8 @@ def log(self, base: ElementLike | ArrayLike | None = None) -> int | np.ndarray:
 
         if np.isscalar(output):
             output = int(output)
+        if output.dtype == np.object_:
+            output = output.astype(int)
 
         return output