fix node type not being parsed correctly in Node constructor

OpenMined · Feb 9, 2024 · 459c112 · 459c112
1 parent 4c38048
commit 459c112
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/packages/hagrid/hagrid/orchestra.py b/packages/hagrid/hagrid/orchestra.py
@@ -259,7 +259,7 @@ def deploy_to_python(
         "processes": processes,
         "dev_mode": dev_mode,
         "tail": tail,
-        "node_type": node_type_enum,
+        "node_type": str(node_type_enum),
         "node_side_type": node_side_type,
         "enable_warnings": enable_warnings,
         # new kwargs

diff --git a/packages/syft/src/syft/abstract_node.py b/packages/syft/src/syft/abstract_node.py
@@ -26,6 +26,9 @@ class NodeSideType(str, Enum):
     LOW_SIDE = "low"
     HIGH_SIDE = "high"
 
+    def __str__(self) -> str:
+        return self.value
+
 
 class AbstractNode:
     id: Optional[UID]

diff --git a/packages/syft/src/syft/node/node.py b/packages/syft/src/syft/node/node.py
@@ -633,6 +633,9 @@ def named(
                 client_config=blob_client_config
             )
 
+        node_type = NodeType(node_type)
+        node_side_type = NodeSideType(node_side_type)
+
         return cls(
             name=name,
             id=uid,