From 3c97e75db6fd4dc5f93514954139079defe0e72f Mon Sep 17 00:00:00 2001 From: LordBaryhobal Date: Wed, 8 Jul 2026 14:43:52 +0200 Subject: [PATCH] feat(parser): allow subscript in type annotations --- midas/parser/python.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/midas/parser/python.py b/midas/parser/python.py index 269000e..b18bb62 100644 --- a/midas/parser/python.py +++ b/midas/parser/python.py @@ -380,7 +380,7 @@ class PythonParser: for col in cols: columns.append(self._parse_frame_column(col)) - case ast.Slice() | ast.Name(): + case ast.Slice() | ast.Name() | ast.Subscript(): columns.append(self._parse_frame_column(schema)) case _: @@ -391,7 +391,7 @@ class PythonParser: def _parse_frame_column(self, column: ast.expr) -> FrameColumn: loc: Location = Location.from_ast(column) match column: - case ast.Name(): + case ast.Name() | ast.Subscript(): return FrameColumn( location=loc, name=None,