py2dag 0.1.11__tar.gz → 0.1.13__tar.gz

{py2dag-0.1.11 → py2dag-0.1.13}/PKG-INFO

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

{py2dag-0.1.11 → py2dag-0.1.13}/py2dag/parser.py

@@ -66,17 +66,32 @@ 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:
                         if kw.arg is None:
                             raise DSLParseError("**kwargs are not allowed")
-                        kwargs[kw.arg] = _literal(kw.value)
+                        # Support variable-name keyword args as dependencies; literals remain in args
+                        if isinstance(kw.value, ast.Name):
+                            name = kw.value.id
+                            if name not in defined:
+                                raise DSLParseError(f"Undefined dependency: {name}")
+                            deps.append(name)
+                        else:
+                            kwargs[kw.arg] = _literal(kw.value)
 
                     ops.append({"id": var_name, "op": op_name, "deps": deps, "args": kwargs})
                 elif isinstance(value, ast.JoinedStr):

{py2dag-0.1.11 → py2dag-0.1.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "py2dag"
-version = "0.1.11"
+version = "0.1.13"
 description = "Convert Python function plans to DAG (JSON, pseudo, optional SVG)."
 authors = ["rvergis"]
 license = "MIT"