computation-graph 57__tar.gz → 58__tar.gz

{computation_graph-57 → computation_graph-58}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: computation-graph
-Version: 57
+Version: 58
 Requires-Python: >=3
 Description-Content-Type: text/markdown
 License-File: LICENSE.md

{computation_graph-57 → computation_graph-58}/computation_graph/graph_test.py

@@ -233,6 +233,17 @@ def test_first():
     )
 
 
+def test_raise_unhandled_exception():
+    class MyExceptionError(Exception):
+        ...
+
+    def raises():
+        raise MyExceptionError("BAD")
+
+    with pytest.raises(MyExceptionError, match="BAD"):
+        graph_runners.nullary_infer_sink(composers.make_first(raises, lambda: 1))
+
+
 def test_first_all_unactionable():
     def raises():
         raise base_types.SkipComputationError

{computation_graph-57 → computation_graph-58}/computation_graph/run.py

@@ -71,7 +71,13 @@ _toposort_nodes: Callable[
         opt_gamla.compose_left(opt_gamla.map(base_types.edge_destination), set)
     ),
     _transpose_graph,
-    lambda g: toposort.toposort_flatten(g, sort=False),
+    toposort.toposort,
+    # Make async functions come first in each layer so they'll start running before all the sync functions
+    opt_gamla.maptuple(
+        gamla.sort_by(lambda n: 0 if inspect.iscoroutinefunction(n.func) else 1)
+    ),
+    gamla.concat,
+    tuple,
 )
 
 

{computation_graph-57 → computation_graph-58}/computation_graph.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: computation-graph
-Version: 57
+Version: 58
 Requires-Python: >=3
 Description-Content-Type: text/markdown
 License-File: LICENSE.md

{computation_graph-57 → computation_graph-58}/setup.py

@@ -7,7 +7,7 @@ with open("README.md", "r") as fh:
 setuptools.setup(
     name="computation-graph",
     python_requires=">=3",
-    version="57",
+    version="58",
     long_description=_LONG_DESCRIPTION,
     long_description_content_type="text/markdown",
     packages=setuptools.find_namespace_packages(),