Also fix for comprehension

sola-st · Aug 28, 2024 · c799d83 · c799d83
1 parent f722cfc
commit c799d83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dynapyt/instrument/CodeInstrumenter.py b/src/dynapyt/instrument/CodeInstrumenter.py
@@ -1894,7 +1894,7 @@ def leave_CompFor(self, original_node, updated_node):
         if (
             "enter_for" not in self.selected_hooks
             and "exit_for" not in self.selected_hooks
-        ):
+        ) or original_node.asynchronous is not None:  # TODO: Handle async for loops
             return updated_node
         generator_name = cst.Attribute(
             value=cst.Name(value="_rt"), attr=cst.Name(value="_gen_")