Fix pt import bug

tempoCollaboration · Oct 7, 2022 · 9313adc · 9313adc
1 parent 4e27eeb
commit 9313adc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/oqupy/process_tensor.py b/oqupy/process_tensor.py
@@ -533,10 +533,10 @@ def _read_file(self, filename: Text):
 
         # transforms
         transform_in = np.array(self._f["transform_in"])
-        if transform_in == 0.0:
+        if np.allclose(transform_in, np.array([0.0])):
             transform_in = None
         transform_out = np.array(self._f["transform_out"])
-        if transform_out == 0.0:
+        if np.allclose(transform_out, np.array([0.0])):
             transform_out = None
 
         # initial tensor and mpo/cap/lam tensors