From cea9f37aad2d861a7aed9840244d5e12912e02ae Mon Sep 17 00:00:00 2001
From: Nathan Freeman <nathan.dean.freeman@gmail.com>
Date: Wed, 3 Jan 2024 16:01:31 -0600
Subject: [PATCH] fix: select args directly from state.ctx

---
 src/engine/src/core/workflows/WorkflowExecutor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/engine/src/core/workflows/WorkflowExecutor.py b/src/engine/src/core/workflows/WorkflowExecutor.py
index 56fcde7e..73e28927 100644
--- a/src/engine/src/core/workflows/WorkflowExecutor.py
+++ b/src/engine/src/core/workflows/WorkflowExecutor.py
@@ -549,7 +549,7 @@ def _prepare_pipeline(self):
 
         # Persist each arg to files in the pipeline file system
         arg_value_file_repo = self.container.load("ArgValueFileRepository")
-        for key in self.state.ctx.pipeline.args:
+        for key in self.state.ctx.args:
             arg_value_file_repo.save(
                 self.state.ctx.pipeline.args_dir + key,
                 self.state.ctx.pipeline.args[key].value