Skip to content

Commit cbb868b

Browse files
fix local-run with args
1 parent 5d748b4 commit cbb868b

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

clarifai/cli/pipeline_step.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def init(pipeline_step_path):
111111
logger.info("4. Implement your pipeline step logic in 1/pipeline_step.py")
112112

113113

114-
@pipeline_step.command('local-run')
114+
@pipeline_step.command(['local-run'])
115115
@click.argument("pipeline_step_path", type=click.Path(exists=True), required=False, default=".")
116116
@click.option(
117117
'--mode',
@@ -172,7 +172,7 @@ def local_run(pipeline_step_path, mode, keep_image, step_args):
172172
manager.run_pipeline_step_container(
173173
image_name=image_name,
174174
container_name=container_name,
175-
step_args=list(step_args) if step_args else None,
175+
step_args=step_args,
176176
)
177177
finally:
178178
if manager.container_exists(container_name):

clarifai/runners/pipeline_steps/pipeline_run_locally.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def run_pipeline_step_container(
110110
cmd.extend(step_args)
111111

112112
logger.info(f"Running pipeline step in container '{container_name}'...")
113-
logger.info(f"Docker command: {cmd}")
113+
logger.info(f"Docker command: {' '.join(cmd)}")
114114

115115
process = subprocess.Popen(cmd)
116116

0 commit comments

Comments
 (0)