fix: passing of input/output paths

HERA-Team · May 16, 2024 · 13d261e · 13d261e
1 parent 03fb4ad
commit 13d261e
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/hera_notebook_templates/cli.py b/hera_notebook_templates/cli.py
@@ -74,16 +74,14 @@ def runfunc(ctx, basename, **kwargs):
         print(f"Executing Notebook and saving to {output_path}")
         print(f"Got notebook params: '{kwargs}'")
 
+        kwargs['papermill_output_path'] = str(output_path)
+        kwargs['papermill_input_path'] = str(nbfile)
+
         pm.execute_notebook(
             str(nbfile),
             output_path = output_path,
             kernel_name = ctx.obj['kernel'],
-            parameters=
-            {
-                'papermill_output_path': str(output_path),
-                'papermill_input_path': str(nbfile),
-                **kwargs,
-            }
+            parameters= kwargs,
         )
 
         for fmt in ctx.obj['formats']: