remove the -y flag as it doesn't exist for pip install (#3058)

zenml-io · Oct 7, 2024 · ca79ab2 · ca79ab2
1 parent 5d0b9eb
commit ca79ab2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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(