Merge pull request #77 from nimamg/fix-list-comp

Fix hook name in CodeInstrumenter
sola-st · Oct 18, 2024 · 9c645af · 9c645af
2 parents b78f8bb + fccceda
commit 9c645af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dynapyt/instrument/CodeInstrumenter.py b/src/dynapyt/instrument/CodeInstrumenter.py
@@ -628,7 +628,7 @@ def leave_List(self, original_node, updated_node):
         return call
 
     def leave_ListComp(self, original_node, updated_node):
-        if "list" not in self.selected_hooks:
+        if "_list" not in self.selected_hooks:
             return updated_node
         callee_name = cst.Attribute(
             value=cst.Name(value="_rt"), attr=cst.Name(value="_list_")