scalar from_json

zxcalc · Sep 26, 2024 · 7bd65b8 · 7bd65b8
1 parent 68e8c68
commit 7bd65b8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pyzx/graph/scalar.py b/pyzx/graph/scalar.py
@@ -194,8 +194,11 @@ def to_json(self) -> str:
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, s: str) -> 'Scalar':
-        d = json.loads(s)
+    def from_json(cls, s: str|dict[str,Any]) -> 'Scalar':
+        if isinstance(s, str):
+            d = json.loads(s)
+        else:
+            d = s
         d["phase"] = Fraction(d["phase"])
         if "phasenodes" in d:
             d["phasenodes"] = [Fraction(p) for p in d["phasenodes"]]