indexify 0.2.26__py3-none-any.whl → 0.2.27__py3-none-any.whl

indexify/functions_sdk/graph.py

@@ -81,7 +81,7 @@ class Graph:
         # Storage for local execution
         self._results: Dict[str, Dict[str, List[IndexifyData]]] = {}
         self._cache = CacheAwareFunctionWrapper("./indexify_local_runner_cache")
-        self._accumulator_values: Dict[str, Dict[str, IndexifyData]] = {}
+        self._accumulator_values: Dict[str, IndexifyData] = {}
         self._local_graph_ctx: Optional[GraphInvocationContext] = None
 
     def get_function(self, name: str) -> IndexifyFunctionWrapper:
@@ -158,7 +158,7 @@ class Graph:
     def definition(self) -> ComputeGraphMetadata:
         start_node = self.nodes[self._start_node]
         is_reducer = False
-        if hasattr(start_node, 'accumulate'):
+        if hasattr(start_node, "accumulate"):
             is_reducer = start_node.accumulate is not None
         start_node = FunctionMetadata(
             name=start_node.name,
@@ -219,13 +219,12 @@ class Graph:
         )
         print(f"[bold] Invoking {self._start_node}[/bold]")
         outputs = defaultdict(list)
-        self._accumulator_values[input.id] = {}
         for k, v in self.accumulator_zero_values.items():
             node = self.nodes[k]
             serializer = get_serializer(node.encoder)
-            self._accumulator_values[input.id] = {
-                k: IndexifyData(payload=serializer.serialize(v), encoder=node.encoder)
-            }
+            self._accumulator_values[k] = IndexifyData(
+                payload=serializer.serialize(v), encoder=node.encoder
+            )
         self._results[input.id] = outputs
         ctx = GraphInvocationContext(
             invocation_id=input.id,
@@ -242,7 +241,6 @@ class Graph:
         initial_input: IndexifyData,
         outputs: Dict[str, List[bytes]],
     ):
-        accumulator_values = self._accumulator_values[initial_input.id]
         queue = deque([(self._start_node, initial_input)])
         while queue:
             node_name, input = queue.popleft()
@@ -258,12 +256,12 @@ class Graph:
             out_edges = self.edges.get(node_name, [])
             fn_outputs = function_outputs.ser_outputs
             print(f"ran {node_name}: num outputs: {len(fn_outputs)}")
-            if accumulator_values.get(node_name, None) is not None:
-                accumulator_values[node_name] = fn_outputs[-1].model_copy()
+            if self._accumulator_values.get(node_name, None) is not None:
+                self._accumulator_values[node_name] = fn_outputs[-1].model_copy()
                 outputs[node_name] = []
             if fn_outputs:
                 outputs[node_name].extend(fn_outputs)
-            if accumulator_values.get(node_name, None) is not None and queue:
+            if self._accumulator_values.get(node_name, None) is not None and queue:
                 print(
                     f"accumulator not none for {node_name}, continuing, len queue: {len(queue)}"
                 )

{indexify-0.2.26.dist-info → indexify-0.2.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: indexify
-Version: 0.2.26
+Version: 0.2.27
 Summary: Python Client for Indexify
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0

{indexify-0.2.26.dist-info → indexify-0.2.27.dist-info}/RECORD

@@ -15,7 +15,7 @@ indexify/executor/runtime_probes.py,sha256=mjw2_mGQ622wRT_39WPGGgPEZQTgtrf3-ICcU
 indexify/executor/task_reporter.py,sha256=7M2fDzLDkSNOSr42EBtkIbGvBzeZlTCqxQuDfdoFT4Y,3349
 indexify/executor/task_store.py,sha256=u48FdRKAh_KH7WOMQOArdOY5CawlyW5MJx8V0W79JM0,3951
 indexify/functions_sdk/data_objects.py,sha256=G08Nrfe2nMUXEhMSedIZwNcZLEsEnj0NykmLPaMZmFs,842
-indexify/functions_sdk/graph.py,sha256=bfVJRAnStBSIxS-6jvOua1lk45RFJjahyzui2OBFcwo,11942
+indexify/functions_sdk/graph.py,sha256=a28o07g0MJoBq2rQWfMGK_dtiSyryoADeF7tEuhI8gc,11817
 indexify/functions_sdk/graph_definition.py,sha256=s8NbqZRaVw1mhGvaaUZ2cJ0NitdlbLj1tnkKjsc71U4,1309
 indexify/functions_sdk/graph_validation.py,sha256=mN2Fcp91GIwFZEQP6z_qGqt4LkLM70SnI7AWBi4CmKQ,2509
 indexify/functions_sdk/image.py,sha256=QK0H6KxLWriB_z4M0kunKzzHdHxYLWL670RPYgYuf_8,1762
@@ -27,8 +27,8 @@ indexify/http_client.py,sha256=mGSSZSMJOShoZGDq42f0kiCQ_aIg2ZjWKeaOW35_r8g,15400
 indexify/remote_graph.py,sha256=_04bdraDaeEmkvEcsxIZ8M37BCAjfbfQccZpAgnZJ2c,4435
 indexify/remote_pipeline.py,sha256=FW7IAv3r24OOpiqlprw3kuFrpdkqi6Ic4_tT26FThjA,761
 indexify/settings.py,sha256=Ny59mzYI4gbXoK8hjx66a_men6ndbd1J1zCTcKOoyzg,50
-indexify-0.2.26.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.26.dist-info/METADATA,sha256=9zbV89Ar2HZqBnC8QCpuiU8Pk1tE8vx1BU3zPf9jGU8,6242
-indexify-0.2.26.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-indexify-0.2.26.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.26.dist-info/RECORD,,
+indexify-0.2.27.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.2.27.dist-info/METADATA,sha256=Ok21hFzjW-T6uEn4fNrZC1ioUeDvmbQdUvlZcYNg7KU,6242
+indexify-0.2.27.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+indexify-0.2.27.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
+indexify-0.2.27.dist-info/RECORD,,