py2dag 0.3.5__tar.gz → 0.3.7__tar.gz

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

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

{py2dag-0.3.5 → py2dag-0.3.7}/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:
@@ -709,11 +722,28 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
         for stmt in body:
             _parse_stmt(stmt)
 
+        # If no explicit return was encountered, emit a terminal break node so
+        # that the plan represents function completion.
+        break_id: Optional[str] = None
+        if returned_var is None:
+            break_id = _ssa_new("break")
+            ops.append({"id": break_id, "op": "CTRL.break", "deps": [], "args": {}})
+
+        # If no outputs were produced, synthesise a default return of `None` so
+        # that parsing succeeds for empty functions.
+        if returned_var is None and not outputs:
+            const_id = _ssa_new("return_value")
+            ops.append({
+                "id": const_id,
+                "op": "CONST.value",
+                "deps": [],
+                "args": {"value": None},
+            })
+            returned_var = const_id
+
         if not outputs:
-            if returned_var is not None:
-                outputs.append({"from": returned_var, "as": "return"})
-            else:
-                raise DSLParseError("At least one output() call required")
+            outputs.append({"from": returned_var if returned_var is not None else break_id, "as": "return"})
+
         if len(ops) > 2000:
             raise DSLParseError("Too many operations")
 

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

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