computation-graph 59__tar.gz → 61__tar.gz

{computation_graph-59 → computation_graph-61}/PKG-INFO

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

{computation_graph-59 → computation_graph-61}/computation_graph/graph_test.py

@@ -233,6 +233,38 @@ def test_first():
     )
 
 
+async def test_raise_handled_from_async():
+    async def raises():
+        raise base_types.SkipComputationError
+
+    assert (
+        await graph_runners.nullary_infer_sink(
+            composers.make_first(raises, lambda: 1, lambda: 2)
+        )
+        == 1
+    )
+
+
+async def test_no_result_for_node_that_raised_handled_exception():
+    async def raises(x):
+        raise base_types.SkipComputationError
+
+    def sink(x):
+        return x
+
+    @graph.make_computation_node
+    def source():
+        return 4
+
+    res = await run.to_callable_strict(composers.compose_left(source, raises, sink))(
+        {}, {}
+    )
+
+    assert res[source] == 4
+    assert raises not in res
+    assert sink not in res
+
+
 def test_raise_unhandled_exception():
     class MyExceptionError(Exception):
         ...

{computation_graph-59 → computation_graph-61}/computation_graph/run.py

@@ -242,9 +242,17 @@ def _to_callable_with_side_effect_for_single_and_multiple(
                     await asyncio.gather(*async_results[1], return_exceptions=True),
                 ):
                     e = node_to_task_or_result[node].exception()
-                    if e:
+                    if not e:
+                        all_results[node] = node_result
+                    elif not isinstance(
+                        e,
+                        (
+                            _DepNotFoundError,
+                            base_types.SkipComputationError,
+                            *handled_exceptions,
+                        ),
+                    ):
                         raise e from e
-                    all_results[node] = node_result
 
             return all_node_side_effects_on_edges(all_results)
 

{computation_graph-59 → computation_graph-61}/computation_graph.egg-info/PKG-INFO

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

{computation_graph-59 → computation_graph-61}/setup.py

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