made fixes

Author: Guy Amram

computation_graph/graph_runners.py

@@ -86,14 +86,15 @@ async def inner(turns):
                 assert (
                     prev[graph.make_computation_node(sink)] == expectation
                 ), f"actual={prev[graph.make_computation_node(sink)]}\n expected: {expectation}"
-    else:
-        def inner(turns):
-            prev = {}
-            for turn, expectation in turns:
-                prev = f(prev, turn)
-                assert (
-                    prev[graph.make_computation_node(sink)] == expectation
-                ), f"actual={prev[graph.make_computation_node(sink)]}\n expected: {expectation}"
+        return inner
+
+    def inner(turns):
+        prev = {}
+        for turn, expectation in turns:
+            prev = f(prev, turn)
+            assert (
+                prev[graph.make_computation_node(sink)] == expectation
+            ), f"actual={prev[graph.make_computation_node(sink)]}\n expected: {expectation}"
 
     return inner