Merge pull request #280 from TTsangSC/issue-279

Added support for annotated assignments in the static parser
pyutils · Jun 24, 2024 · 13bf68f · 13bf68f
2 parents c715356 + ddf3fec
commit 13bf68f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/line_profiler/autoprofile/util_static.py b/line_profiler/autoprofile/util_static.py
@@ -181,6 +181,10 @@ def visit_Assign(self, node):
                 if getattr(target, "id", None) == varname:
                     self.static_value = _parse_static_node_value(node.value)
 
+        def visit_AnnAssign(self, node):
+            if getattr(node.target, "id", None) == varname:
+                self.static_value = _parse_static_node_value(node.value)
+
     visitor = StaticVisitor()
     visitor.visit(pt)
     try: