diff --git a/python-sdk/indexify/functions_sdk/graph.py b/python-sdk/indexify/functions_sdk/graph.py index 3df3ba6bd..5f80c0daf 100644 --- 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 index 07c120d1d..2816ff2b1 100644 --- 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):