py2dag 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

py2dag/parser.py

@@ -66,11 +66,19 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
 
                     deps: List[str] = []
                     for arg in value.args:
-                        if not isinstance(arg, ast.Name):
-                            raise DSLParseError("Positional args must be variable names")
-                        if arg.id not in defined:
-                            raise DSLParseError(f"Undefined dependency: {arg.id}")
-                        deps.append(arg.id)
+                        if isinstance(arg, ast.Name):
+                            if arg.id not in defined:
+                                raise DSLParseError(f"Undefined dependency: {arg.id}")
+                            deps.append(arg.id)
+                        elif isinstance(arg, (ast.List, ast.Tuple)):
+                            for elt in arg.elts:
+                                if not isinstance(elt, ast.Name):
+                                    raise DSLParseError("List/Tuple positional args must be variable names")
+                                if elt.id not in defined:
+                                    raise DSLParseError(f"Undefined dependency: {elt.id}")
+                                deps.append(elt.id)
+                        else:
+                            raise DSLParseError("Positional args must be variable names or lists/tuples of names")
 
                     kwargs: Dict[str, Any] = {}
                     for kw in value.keywords:

{py2dag-0.1.12.dist-info → py2dag-0.1.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: py2dag
-Version: 0.1.12
+Version: 0.1.13
 Summary: Convert Python function plans to DAG (JSON, pseudo, optional SVG).
 License: MIT
 Author: rvergis

{py2dag-0.1.12.dist-info → py2dag-0.1.13.dist-info}/RECORD

@@ -2,10 +2,10 @@ py2dag/__init__.py,sha256=i8VB44JCVcRJAcvnQtbH8YVRUz5j7dE355iRbikXPGQ,250
 py2dag/cli.py,sha256=BCBi5mNxOqeEN8uEMt_hiDx0iSt7ZE3y74cGXREzZ2I,1296
 py2dag/export_dagre.py,sha256=S244wUxBMuM9qXD8bklaslkMp3rcbBGWvMhwWVBzBF0,3487
 py2dag/export_svg.py,sha256=YyjqOuj8GhUTDWP70SKnnSWAKI1PvJwyOhHLwB29uNM,1812
-py2dag/parser.py,sha256=1udeSmeFlH9qSXx5RTjrPg6Uof3_TtizSIMPJasp4aM,10950
+py2dag/parser.py,sha256=AdC76fKiLojxrc7Tk20CHgS7U0ofQwBYecMHQycc3PQ,11507
 py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
-py2dag-0.1.12.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
-py2dag-0.1.12.dist-info/METADATA,sha256=o7jsa9S5EDuqMgvfYBgQ4sBr3kAgVSSrduP08ou27MU,3550
-py2dag-0.1.12.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-py2dag-0.1.12.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
-py2dag-0.1.12.dist-info/RECORD,,
+py2dag-0.1.13.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
+py2dag-0.1.13.dist-info/METADATA,sha256=gG64U7iwhT-lWu84Tu9c9TFTID6jIBHHI184tGXpkqY,3550
+py2dag-0.1.13.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+py2dag-0.1.13.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
+py2dag-0.1.13.dist-info/RECORD,,