Traceback on expression list assignment
Closed this issue · 0 comments
ericwb commented
Describe the bug
Get an IndexError: list index out of range
when parsing code with a specific expression list assignment.
To Reproduce
import torch
torch.tensor([[0.1, 1.2], [2.2, 3.1], [4.9, 5.2]])
x = torch.tensor([[0.1, 1.2], [2.2, 3.1], [4.9, 5.2]])
b, *_, device = *x.shape, x.device
self.visit(nodes)
File "/Users/ericwb/workspace/precli/precli/parsers/__init__.py", line 119, in visit
visitor_fn(node.children)
File "/Users/ericwb/workspace/precli/precli/parsers/python.py", line 48, in visit_class_definition
self.visit(nodes)
File "/Users/ericwb/workspace/precli/precli/parsers/__init__.py", line 119, in visit
visitor_fn(node.children)
File "/Users/ericwb/workspace/precli/precli/parsers/__init__.py", line 119, in visit
visitor_fn(node.children)
File "/Users/ericwb/workspace/precli/precli/parsers/__init__.py", line 119, in visit
visitor_fn(node.children)
File "/Users/ericwb/workspace/precli/precli/parsers/python.py", line 55, in visit_function_definition
self.visit(nodes)
File "/Users/ericwb/workspace/precli/precli/parsers/__init__.py", line 119, in visit
visitor_fn(node.children)
File "/Users/ericwb/workspace/precli/precli/parsers/__init__.py", line 119, in visit
visitor_fn(node.children)
File "/Users/ericwb/workspace/precli/precli/parsers/__init__.py", line 119, in visit
visitor_fn(node.children)
File "/Users/ericwb/workspace/precli/precli/parsers/python.py", line 102, in visit_assignment
nodes[2].named_children[i],
~~~~~~~~~~~~~~~~~~~~~~~^^^
IndexError: list index out of range
Expected behavior
No exceptions
Version
precli 0.4.4.dev5
python version = 3.12.1 (main, Dec 12 2023, 13:19:17) [Clang 15.0.0 (clang-1500.0.40.1)]
Additional context
n/a