update: node to require bytes to instantiate node class

olivmath · Dec 31, 2023 · e41ff56 · e41ff56
1 parent 82ddc61
commit e41ff56
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/merkly/node.py b/merkly/node.py
@@ -1,6 +1,6 @@
-from enum import Enum
+from pydantic import BaseModel, StrictBytes
 from typing import Optional
-from pydantic import BaseModel
+from enum import Enum
 
 
 class Side(Enum):
@@ -13,7 +13,7 @@ class Node(BaseModel):
     # 🍃 Leaf of Merkle Tree
     """
 
-    data: Optional[bytes] = None
+    data: Optional[StrictBytes] = None
     side: Side = Side.LEFT
 
     def __eq__(self, other: "Node") -> bool: