From 9313adce85703a9abefa0c9232f97ed9aa7e67a6 Mon Sep 17 00:00:00 2001
From: "Gerald E. Fux" <gerald.e.fux@gmail.com>
Date: Fri, 7 Oct 2022 00:11:28 +0200
Subject: [PATCH] Fix pt import bug

---
 oqupy/process_tensor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/oqupy/process_tensor.py b/oqupy/process_tensor.py
index b80569fa..d37f05e5 100644
--- 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