diff --git a/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator.py b/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator.py index 82366bbdd9..73b73410ad 100644 --- a/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator.py +++ b/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator.py @@ -399,7 +399,7 @@ def _construct_lightning_steps( f"Installing requirements: {pipeline_requirements_to_string}" ) studio.run(f"uv pip install {pipeline_requirements_to_string}") - studio.run("pip install zenml -y") + studio.run("pip install zenml") for custom_command in settings.custom_commands or []: studio.run( @@ -493,7 +493,7 @@ def _upload_and_run_pipeline( ) studio.run("pip install uv") studio.run(f"uv pip install {requirements}") - studio.run("pip install zenml -y") + studio.run("pip install zenml") # studio.run(f"pip install {wheel_path.rsplit('/', 1)[-1]}") for command in settings.custom_commands or []: output = studio.run( @@ -566,7 +566,7 @@ def _run_step_in_new_studio( ) studio.run("pip install uv") studio.run(f"uv pip install {details['requirements']}") - studio.run("pip install zenml -y") + studio.run("pip install zenml") # studio.run(f"pip install {wheel_path.rsplit('/', 1)[-1]}") for command in custom_commands or []: output = studio.run( diff --git a/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator_entrypoint.py b/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator_entrypoint.py index 4f0ea6394f..f3d914070b 100644 --- a/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator_entrypoint.py +++ b/src/zenml/integrations/lightning/orchestrators/lightning_orchestrator_entrypoint.py @@ -166,7 +166,7 @@ def main() -> None: f"uv pip install {pipeline_requirements_to_string}" ) logger.info(output) - output = main_studio.run("pip install zenml -y") + output = main_studio.run("pip install zenml") logger.info(output) for command in pipeline_settings.custom_commands or []: @@ -248,7 +248,7 @@ def run_step_on_lightning_studio(step_name: str) -> None: f"uv pip install {step_requirements_to_string}" ) logger.info(output) - output = studio.run("pip install zenml -y") + output = studio.run("pip install zenml") logger.info(output) for command in step_settings.custom_commands or []: output = studio.run(