sera-2 1.21.1__py3-none-any.whl → 1.21.2__py3-none-any.whl

sera/libs/directed_computing_graph/_dcg.py

@@ -41,6 +41,11 @@ class DirectedComputingGraph:
     ):
         self.graph = graph
         self.type_service = type_service
+        self.node2descendants: dict[str, list[DCGNode]] = {}
+
+        for u in graph.iter_nodes():
+            self.node2descendants[u.id] = graph.descendants(u.id)
+            self.node2descendants[u.id].append(u)
 
     @staticmethod
     def from_flows(
@@ -179,7 +184,7 @@ class DirectedComputingGraph:
     def execute(
         self,
         input: dict[ComputeFnId, tuple],
-        output: set[str],
+        output: Optional[set[str]] = None,
         context: Optional[
             dict[str, Callable | Any] | Callable[[], dict[str, Any]]
         ] = None,
@@ -203,6 +208,9 @@ class DirectedComputingGraph:
         else:
             context = {k: v() if callable(v) else v for k, v in context.items()}
 
+        if output is None:
+            output = set()
+
         # This is a quick reactive algorithm, we may be able to do it better.
         # The idea is when all inputs of a function is available, we can execute a function.
         # We assume that the memory is large enough to hold all the functions and their inputs
@@ -211,25 +219,25 @@ class DirectedComputingGraph:
         # we execute the computing nodes
         # when it's finished, we put the outgoing edges into a stack.
         runtimes: dict[NodeId, NodeRuntime] = {}
+        for id in input.keys():
+            for u in self.node2descendants[id]:
+                if u.id in input:
+                    # user provided input should supersede the context
+                    n_provided_args = len(input[u.id])
+                    n_consumed_context = n_provided_args - len(u.required_args)
+                else:
+                    n_consumed_context = 0
 
-        for u in self.graph.iter_nodes():
-            if u.id in input:
-                # user provided input should supersede the context
-                n_provided_args = len(input[u.id])
-                n_consumed_context = n_provided_args - len(u.required_args)
-            else:
-                n_consumed_context = 0
-
-            node_context = tuple(
-                (
-                    context[name]
-                    if name in context
-                    else u.required_context_default_args[name]
+                node_context = tuple(
+                    (
+                        context[name]
+                        if name in context
+                        else u.required_context_default_args[name]
+                    )
+                    for name in u.required_context[n_consumed_context:]
                 )
-                for name in u.required_context[n_consumed_context:]
-            )
 
-            runtimes[u.id] = NodeRuntime.from_node(self.graph, u, node_context)
+                runtimes[u.id] = NodeRuntime.from_node(self.graph, u, node_context)
         stack: list[NodeId] = []
 
         for id, args in input.items():
@@ -321,24 +329,25 @@ class DirectedComputingGraph:
         # when it's finished, we put the outgoing edges into a stack.
         runtimes: dict[NodeId, NodeRuntime] = {}
 
-        for u in self.graph.iter_nodes():
-            if u.id in input:
-                # user provided input should supersede the context
-                n_provided_args = len(input[u.id])
-                n_consumed_context = n_provided_args - len(u.required_args)
-            else:
-                n_consumed_context = 0
+        for id in input.keys():
+            for u in self.node2descendants[id]:
+                if u.id in input:
+                    # user provided input should supersede the context
+                    n_provided_args = len(input[u.id])
+                    n_consumed_context = n_provided_args - len(u.required_args)
+                else:
+                    n_consumed_context = 0
 
-            node_context = tuple(
-                (
-                    context[name]
-                    if name in context
-                    else u.required_context_default_args[name]
+                node_context = tuple(
+                    (
+                        context[name]
+                        if name in context
+                        else u.required_context_default_args[name]
+                    )
+                    for name in u.required_context[n_consumed_context:]
                 )
-                for name in u.required_context[n_consumed_context:]
-            )
+                runtimes[u.id] = NodeRuntime.from_node(self.graph, u, node_context)
 
-            runtimes[u.id] = NodeRuntime.from_node(self.graph, u, node_context)
         stack: list[NodeId] = []
 
         for id, args in input.items():

sera/misc/_utils.py

@@ -417,3 +417,5 @@ async def replay_events(
                 await dcg.execute_async(
                     input={innode: (record,)}, context={"session": session}
                 )
+
+        await session.commit()

{sera_2-1.21.1.dist-info → sera_2-1.21.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sera-2
-Version: 1.21.1
+Version: 1.21.2
 Summary: 
 Author: Binh Vu
 Author-email: bvu687@gmail.com

{sera_2-1.21.1.dist-info → sera_2-1.21.2.dist-info}/RECORD

@@ -9,7 +9,7 @@ sera/libs/api_test_helper.py,sha256=3tRr8sLN4dBSrHgKAHMmyoENI0xh7K_JLel8AvujU7k,
 sera/libs/base_orm.py,sha256=5hOH_diUeaABm3cpE2-9u50VRqG1QW2osPQnvVHIhIA,3365
 sera/libs/base_service.py,sha256=NY-PLiaS1SXs2Pp3yVsdGnpJGUxKbHb0qzCCguhgPP4,6432
 sera/libs/directed_computing_graph/__init__.py,sha256=xiF5_I1y9HtQ-cyq02iwkRYgEZvxBB8YIvysCHCLBco,1290
-sera/libs/directed_computing_graph/_dcg.py,sha256=miclcxik67CbJpb8slvUOcdWLQH67msqder7JpOoiTw,14870
+sera/libs/directed_computing_graph/_dcg.py,sha256=nQf9MhnTkFU2-dxhv_PFThz9L61Nn8cvYIOoMq0OVb8,15352
 sera/libs/directed_computing_graph/_edge.py,sha256=iBq6cpLWWyuD99QWTHVEh8naWUJrR4WJJuq5iuCrwHo,1026
 sera/libs/directed_computing_graph/_flow.py,sha256=6v39yKPIDYrQ3KvFqjeAWs88-oQSnDTaED2F3LF2z_I,478
 sera/libs/directed_computing_graph/_fn_signature.py,sha256=73iPUITcRKW0-l6sqjwMSk_FZnJESaKOmUKDGHTOh9Q,1598
@@ -28,7 +28,7 @@ sera/make/make_python_services.py,sha256=0ZpWLwQ7Nwfn8BXAikAB4JRpNknpSJyJgY5b1cj
 sera/make/make_typescript_model.py,sha256=CZc_LMigy7e6tP-6bJ5gHZlHYeyRtU_CNR4s64x51xE,71369
 sera/misc/__init__.py,sha256=L70kdXM5WmtMpgLlQISnGmrLgn241TQF0zFXFPi3Xzk,727
 sera/misc/_formatter.py,sha256=aCGYL08l8f3aLODHxSocxBBwkRYEo3K1QzCDEn3suj0,1685
-sera/misc/_utils.py,sha256=7vDhG1_hEdA4enTtL6gg5evBEg397kuxZ5ZtXC9PBWo,12939
+sera/misc/_utils.py,sha256=8J-2z_3n6VQB8DuDrm7wKEq880x3BZLz9kbukORMuas,12971
 sera/models/__init__.py,sha256=vJC5Kzo_N7wd16ocNPy1VvAZDGNiWeiAhWJ4ihATKvA,780
 sera/models/_class.py,sha256=1J4Bd_LanzhhDWwZFHWGtFYD7lupe_alaB3D02ebNDI,2862
 sera/models/_collection.py,sha256=nLlP85OfEhfj4pFAYxB5ZvtBN6EdPSWPJ5ZFh5SSEC4,2686
@@ -42,6 +42,6 @@ sera/models/_parse.py,sha256=9iaW-8Ajq8e440FrTJ-X9oivHJJ447kPU-ZqffFxSvw,12649
 sera/models/_property.py,sha256=9yMDxrmbyuF6-29lQjiq163Xzwbk75TlmGBpu0NLpkI,7485
 sera/models/_schema.py,sha256=VxJEiqgVvbXgcSUK4UW6JnRcggk4nsooVSE6MyXmfNY,1636
 sera/typing.py,sha256=m4rir-fB6Cgcm7_ZSXXcNdla2LJgq96WXxtTTrDaJno,1058
-sera_2-1.21.1.dist-info/METADATA,sha256=ojxFGoSzu_QLUSvtKfl6pE86Qd2eCg6QStYYiFbykGw,936
-sera_2-1.21.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-sera_2-1.21.1.dist-info/RECORD,,
+sera_2-1.21.2.dist-info/METADATA,sha256=8jE5v1c-64MjGwTRTiPto2OD5tjYlZEkGtkmMkyZ2tk,936
+sera_2-1.21.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+sera_2-1.21.2.dist-info/RECORD,,