Merge pull request #44 from hyroai/allow-compose_unary-to-a-graph-destination

allow compose_unary to a graph destination

Author: molaxx

computation_graph/composers/__init__.py

@@ -217,7 +217,9 @@ def _infer_composition_edges(
             gamla.unique,
             gamla.filter(
                 gamla.compose_left(
-                    unbound_signature, lambda signature: key in signature.kwargs
+                    unbound_signature,
+                    lambda sig: key in sig.kwargs
+                    or (key is None and signature.is_unary(sig)),
                 )
             ),
             # Do not add edges to nodes from source that are already present in destination (cycle).

computation_graph/composers/composers_test.py

@@ -1,6 +1,16 @@
 from computation_graph import composers, graph_runners
 
 
+def test_unary_composition_with_graph_destination():
+    result = graph_runners.nullary_infer_sink(
+        composers.compose_unary(
+            composers.compose_left(lambda: 1, lambda a, b: a - b, key="a"), lambda: 2
+        )
+    )
+
+    assert result == -1
+
+
 def test_ambiguity_does_not_blow_up():
     counter = 0
 

setup.py

@@ -7,7 +7,7 @@
 setuptools.setup(
     name="computation-graph",
     python_requires=">=3",
-    version="36",
+    version="37",
     long_description=_LONG_DESCRIPTION,
     long_description_content_type="text/markdown",
     packages=setuptools.find_namespace_packages(),