fix graph definition

tensorlakeai · Nov 6, 2024 · c88dc6f · c88dc6f
1 parent 0d5471a
commit c88dc6f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python-sdk/indexify/functions_sdk/graph.py b/python-sdk/indexify/functions_sdk/graph.py
@@ -160,7 +160,7 @@ def definition(self) -> ComputeGraphMetadata:
             reducer=start_node.accumulate is not None,
             image_name=start_node.image._image_name,
             image_information=start_node.image.to_image_information(),
-            encoder=start_node.encoder
+            payload_encoder=start_node.encoder
         )
         metadata_edges = self.edges.copy()
         metadata_nodes = {}
@@ -172,7 +172,7 @@ def definition(self) -> ComputeGraphMetadata:
                         description=node.description or "",
                         source_fn=node_name,
                         target_fns=self.routers[node_name],
-                        encoder=node.encoder,
+                        payload_encoder=node.encoder,
                         image_name=node.image._image_name,
                         image_information=node.image.to_image_information(),
                     )

diff --git a/python-sdk/indexify/functions_sdk/graph_definition.py b/python-sdk/indexify/functions_sdk/graph_definition.py
@@ -14,7 +14,7 @@ class FunctionMetadata(BaseModel):
     reducer: bool = False
     image_name: str
     image_information: ImageInformation
-    encoder: str = "cloudpickle"
+    payload_encoder: str = "cloudpickle"
 
 
 class RouterMetadata(BaseModel):
@@ -24,7 +24,7 @@ class RouterMetadata(BaseModel):
     target_fns: List[str]
     image_name: str
     image_information: ImageInformation
-    encoder: str = "cloudpickle"
+    payload_encoder: str = "cloudpickle"
 
 
 class NodeMetadata(BaseModel):