Fix np.mod() in NumPy v2

mhostetter · Jun 22, 2024 · f189c95 · f189c95
1 parent 6189c1f
commit f189c95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/galois/_domains/_ufunc.py b/src/galois/_domains/_ufunc.py
@@ -394,7 +394,7 @@ def __call__(self, ufunc, method, inputs, kwargs, meta):
  self._verify_operands_in_field_or_int(ufunc, inputs, meta)
  inputs, kwargs = self._view_inputs_as_ndarray(inputs, kwargs)
  i = meta["non_field_operands"][0] # Scalar multiplicand
- inputs[i] = np.mod(inputs[i], self.field.characteristic)
+ inputs[i] = np.mod(inputs[i], self.field.characteristic, dtype=inputs[i].dtype)
  inputs, kwargs = self._view_inputs_as_ndarray(inputs, kwargs)
  output = getattr(self.ufunc, method)(*inputs, **kwargs)
  output = self._view_output_as_field(output, self.field, meta["dtype"])