Fix dtype for bool

tum-pbs · Nov 24, 2023 · f0f4086 · f0f4086
1 parent b120e4d
commit f0f4086
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/phiml/backend/_numpy_backend.py b/phiml/backend/_numpy_backend.py
@@ -393,6 +393,8 @@ def ifft(self, k, axes: Union[tuple, list]):
             return np.fft.ifftn(k, axes=axes).astype(k.dtype)
 
     def dtype(self, array) -> DType:
+        if isinstance(array, bool):
+            return DType(bool)
         if isinstance(array, int):
             return DType(int, 32)
         if isinstance(array, float):

diff --git a/phiml/backend/jax/_jax_backend.py b/phiml/backend/jax/_jax_backend.py
@@ -564,6 +564,8 @@ def ifft(self, k, axes: Union[tuple, list]):
             return jnp.fft.ifftn(k, axes=axes).astype(k.dtype)
 
     def dtype(self, array) -> DType:
+        if isinstance(array, bool):
+            return DType(bool)
         if isinstance(array, int):
             return DType(int, 32)
         if isinstance(array, float):