Ensure enums can be serialized.

lynxkite-app/tests/test_crdt.py

@@ -18,7 +18,7 @@ def empty_list_workspace():
     yield ws
 
 
-class MyEnum(Enum):
+class MyEnum(int, Enum):
     VALUE = 1
 
 

lynxkite-core/src/lynxkite/core/ops.py

@@ -95,7 +95,7 @@ class ParameterGroup(BaseConfig):
     type: str = "group"
 
 
-class Position(enum.Enum):
+class Position(str, enum.Enum):
     """Defines the position of an input or output in the UI."""
 
     LEFT = "left"

lynxkite-core/src/lynxkite/core/workspace.py

@@ -175,8 +175,7 @@ class Workspace(BaseConfig):
                     data.meta = op
                     # If the node is connected to a CRDT, update that too.
                     if hasattr(node, "_crdt"):
-                        # Go through JSON to simplify the types. CRDT can't handle enums.
-                        node._crdt["data"]["meta"] = json.loads(op.model_dump_json())
+                        node._crdt["data"]["meta"] = op.model_dump()
                         print("set metadata to", op)
                 if node.type != op.type:
                     node.type = op.type

lynxkite-core/tests/test_ops.py

@@ -54,7 +54,7 @@ def test_op_decorator_with_params_and_types_():
 
 
 def test_op_decorator_with_complex_types():
-    class Color(enum.Enum):
+    class Color(int, enum.Enum):
         RED = 1
         GREEN = 2
         BLUE = 3

lynxkite-graph-analytics/src/lynxkite_graph_analytics/ml_ops.py

@@ -148,7 +148,7 @@ VIRIDIS = [
 ]
 
 
-class UMAPMetric(enum.Enum):
+class UMAPMetric(str, enum.Enum):
     l1 = "l1"
     cityblock = "cityblock"
     taxicab = "taxicab"

lynxkite-graph-analytics/src/lynxkite_graph_analytics/pytorch/pytorch_ops.py

@@ -65,7 +65,7 @@ def linear(x, *, output_dim=1024):
     return pyg_nn.Linear(-1, output_dim)
 
 
-class ActivationTypes(enum.Enum):
+class ActivationTypes(str, enum.Enum):
     ReLU = "ReLU"
     Leaky_ReLU = "Leaky ReLU"
     Tanh = "Tanh"