py2dag 0.3.5__tar.gz → 0.3.6__tar.gz

{py2dag-0.3.5 → py2dag-0.3.6}/PKG-INFO

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

{py2dag-0.3.5 → py2dag-0.3.6}/py2dag/parser.py

@@ -131,7 +131,7 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                         return list(prev.get("deps", []))
                 return [ssa_var]
 
-            for arg in call.args:
+            for idx, arg in enumerate(call.args):
                 if isinstance(arg, ast.Starred):
                     star_val = arg.value
                     if isinstance(star_val, ast.Name):
@@ -156,7 +156,20 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                         deps.append(_ssa_get(elt.id))
                         dep_labels.append("")
                 else:
-                    raise DSLParseError("Positional args must be variable names or lists/tuples of names")
+                    # Allow literal positional arguments by emitting a CONST node
+                    try:
+                        lit = _literal(arg)
+                    except DSLParseError:
+                        raise DSLParseError("Positional args must be variable names or lists/tuples of names or literals")
+                    const_id = _ssa_new(f"{var_name}_arg{idx}")
+                    ops.append({
+                        "id": const_id,
+                        "op": "CONST.value",
+                        "deps": [],
+                        "args": {"value": lit},
+                    })
+                    deps.append(const_id)
+                    dep_labels.append("")
 
             kwargs: Dict[str, Any] = {}
             for kw in call.keywords:

{py2dag-0.3.5 → py2dag-0.3.6}/pyproject.toml

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