Fixed Realizer to follow consistent real/compelx behavior

749e909e · Reimar Leike · Philipp Arras · f2c30d01 · 749e909e
Commit 749e909e authored May 25, 2020 by Reimar Leike Committed by Philipp Arras Jun 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

src/operators/simple_linear_operators.py src/operators/simple_linear_operators.py +7 -1

No files found.
--- a/src/operators/simple_linear_operators.py
+++ b/src/operators/simple_linear_operators.py
@@ -109,7 +109,13 @@ class Realizer(EndomorphicOperator):

    def apply(self, x, mode):
        self._check_input(x, mode)
-        return x.real
+        if mode == self.TIMES:
+            if not np.issubdtype(x.dtype, np.complexfloating):
+                raise ValueError
+            return x.real
+        if x.dtype not in (np.float64, np.float32):
+            raise ValueError
+        return Field(x.domain, x.val.astype(np.complex128))


 class Imaginizer(EndomorphicOperator):