Fix ml

Quantco · Nov 1, 2024 · 3c8e9b1 · 3c8e9b1
1 parent 1d6bcbc
commit 3c8e9b1
Show file tree

Hide file tree

Showing 3 changed files with 227 additions and 172 deletions.
diff --git a/src/spox/_public.py b/src/spox/_public.py
@@ -303,7 +303,9 @@ def inline_inner(*args: Var, **kwargs: Var) -> dict[str, Var]:
             model=model,
         )
 
-        return node.get_output_vars(flatten_variadic=True)
+        return dict(
+            zip(out_names, node.get_output_vars(flatten_variadic=True).values())
+        )
 
     return inline_inner