py2dag 0.1.10__tar.gz → 0.1.12__tar.gz

{py2dag-0.1.10 → py2dag-0.1.12}/PKG-INFO

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

{py2dag-0.1.10 → py2dag-0.1.12}/py2dag/parser.py

@@ -76,7 +76,14 @@ def parse(source: str, function_name: Optional[str] = None) -> 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):
@@ -171,9 +178,11 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                     returned_var = const_id
                 else:
                     raise DSLParseError("return must return a variable name or literal")
-            elif isinstance(stmt, (ast.For, ast.AsyncFor, ast.While)):
+            elif isinstance(stmt, (ast.For, ast.AsyncFor, ast.While, ast.If)):
                 # Ignore control flow blocks; only top-level linear statements are modeled
                 continue
+            elif isinstance(stmt, (ast.Pass,)):
+                continue
             else:
                 raise DSLParseError("Only assignments, expression calls, and a final return are allowed in function body")
 

{py2dag-0.1.10 → py2dag-0.1.12}/pyproject.toml

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