Fix lazy

PrefectHQ · May 17, 2024 · e577a4a · e577a4a
1 parent 7c8c91a
commit e577a4a
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/src/controlflow/decorators.py b/src/controlflow/decorators.py
@@ -96,13 +96,14 @@ def wrapped_flow(*args, lazy_=None, **kwargs):
                 with controlflow.instructions(instructions):
                     result = fn(*args, **kwargs)
 
-                    # determine if we should run the flow eagerly or lazily
+                    # Determine if we should run eagerly or lazily
                     if lazy_ is not None:
-                        lazy = lazy_
-                    if lazy is None:
-                        run_eagerly = controlflow.settings.eager_mode
-                    else:
+                        run_eagerly = not lazy_
+                    elif lazy is not None:
                         run_eagerly = not lazy
+                    else:
+                        run_eagerly = controlflow.settings.eager_mode
+
                     if run_eagerly:
                         flow_obj.run()
 
@@ -194,13 +195,14 @@ def wrapper(*args, lazy_: bool = None, **kwargs):
             tools=tools or [],
         )
 
-        # determine if we should run the flow eagerly or lazily
+        # Determine if we should run eagerly or lazily
         if lazy_ is not None:
-            lazy = lazy_
-        if lazy is None:
-            run_eagerly = controlflow.settings.eager_mode
-        else:
+            run_eagerly = not lazy_
+        elif lazy is not None:
             run_eagerly = not lazy
+        else:
+            run_eagerly = controlflow.settings.eager_mode
+
         if run_eagerly:
             task.run()
             return task.result