zenml-nightly 0.58.0.dev20240529__py3-none-any.whl → 0.58.0.dev20240531__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.58.0.dev20240529
+0.58.0.dev20240531

zenml/config/docker_settings.py

@@ -128,6 +128,8 @@ class DockerSettings(BaseSettings):
             therefore **not** include any registry.
         python_package_installer: The package installer to use for python
             packages.
+        python_package_installer_args: Arguments to pass to the python package
+            installer.
         replicate_local_python_environment: If not `None`, ZenML will use the
             specified method to generate a requirements file that replicates
             the packages installed in the currently running python environment.
@@ -185,6 +187,7 @@ class DockerSettings(BaseSettings):
     python_package_installer: PythonPackageInstaller = (
         PythonPackageInstaller.PIP
     )
+    python_package_installer_args: Dict[str, Any] = {}
     replicate_local_python_environment: Optional[
         Union[List[str], PythonEnvironmentExportMethod]
     ] = None

zenml/utils/pipeline_docker_image_builder.py

@@ -20,6 +20,7 @@ import sys
 from collections import defaultdict
 from typing import (
     TYPE_CHECKING,
+    Any,
     DefaultDict,
     Dict,
     List,
@@ -65,6 +66,14 @@ DEFAULT_DOCKER_PARENT_IMAGE = (
     f"py{sys.version_info.major}.{sys.version_info.minor}"
 )
 
+PIP_DEFAULT_ARGS = {
+    "no-cache-dir": None,
+    "default-timeout": 60,
+}
+UV_DEFAULT_ARGS = {
+    "no-cache-dir": None,
+}
+
 
 class PipelineDockerImageBuilder:
     """Builds Docker images to run a ZenML pipeline."""
@@ -636,22 +645,32 @@ class PipelineDockerImageBuilder:
             docker_settings.python_package_installer
             == PythonPackageInstaller.PIP
         ):
-            install_command = "pip install --default-timeout=60"
+            install_command = "pip install"
+            default_installer_args: Dict[str, Any] = PIP_DEFAULT_ARGS
         elif (
             docker_settings.python_package_installer
             == PythonPackageInstaller.UV
         ):
             lines.append("RUN pip install uv")
-            install_command = "uv pip install --system"
+            install_command = "uv pip install"
+            default_installer_args = UV_DEFAULT_ARGS
         else:
             raise ValueError("Unsupported python package installer.")
 
+        installer_args = {
+            **default_installer_args,
+            **docker_settings.python_package_installer_args,
+        }
+        installer_args_string = " ".join(
+            f"--{key}" if value is None else f"--{key}={value}"
+            for key, value in installer_args.items()
+        )
         for file, _, options in requirements_files:
             lines.append(f"COPY {file} .")
             option_string = " ".join(options)
 
             lines.append(
-                f"RUN {install_command} --no-cache-dir "
+                f"RUN {install_command} {installer_args_string}"
                 f"{option_string} -r {file}"
             )
 

{zenml_nightly-0.58.0.dev20240529.dist-info → zenml_nightly-0.58.0.dev20240531.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zenml-nightly
-Version: 0.58.0.dev20240529
+Version: 0.58.0.dev20240531
 Summary: ZenML: Write production-ready ML code.
 Home-page: https://zenml.io
 License: Apache-2.0

{zenml_nightly-0.58.0.dev20240529.dist-info → zenml_nightly-0.58.0.dev20240531.dist-info}/RECORD

@@ -6,7 +6,7 @@ RELEASE_NOTES.md,sha256=0XrQIc5jtfcHBi9zST-f4zdPs9M-YzNRtFRtpYTWq6k,326337
 ROADMAP.md,sha256=hiLSmr16BH8Dfx7SaQM4JcXCGCVl6mFZPFAwJeDTrJU,407
 SECURITY.md,sha256=9DepA8y03yvCZLHEfcXLTDH4lUyKHquAdukBsccNN7c,682
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=1XPr9n6r8gEt8W2QvCEPNZmak_pQcUSbzr3ECUR_Vwk,19
+zenml/VERSION,sha256=lRkGs_nDfVij0xosQ1OFEtObEI3mh87QYpo8ZmPAWCE,19
 zenml/__init__.py,sha256=qT3QHiuTgUwk05sOiML0Gcjre3pugvDAw84_br9Psvo,2394
 zenml/_hub/__init__.py,sha256=6qDzpQAAZa__Aiiz0mC1qM-9dw9_jk_v_aXeJknxbDE,644
 zenml/_hub/client.py,sha256=LAxLuDjY9ya1PPRzxS0_pvu__FSpAqEjbtHTqVwgMWc,9071
@@ -80,7 +80,7 @@ zenml/config/base_settings.py,sha256=y_i95RBfgCo7CnqG0FTwQXQ0rwc9rbl8WClc3X7EvoY
 zenml/config/build_configuration.py,sha256=uFo9LAkCbfX1HTX3RYX8412w4ifbdElV7SP1CvhIHfI,4304
 zenml/config/compiler.py,sha256=x2kgecp6oMFYtWtduwmlnidUMBHTFSO9FN1OtNSPZ5s,22261
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
-zenml/config/docker_settings.py,sha256=CGCCf8Fj0LLhQi2S-7HzGnIxzkcmkMSHlzaQWmwWMPM,11859
+zenml/config/docker_settings.py,sha256=1dh0VhcopKaiQ5GqqmUt9Q22STzICzWaRubmZ-xbEpE,12016
 zenml/config/global_config.py,sha256=JlC3UXHeHIJa6TWlOrxneH1FsraHs_DLpDdlqUfRLE8,29409
 zenml/config/pipeline_configurations.py,sha256=oqP-bYLAGlnM-bAdaRijXtImkGY8keoX2tGgHQwg7PM,2992
 zenml/config/pipeline_run_configuration.py,sha256=y7_yJ9193C-zNSIe6yZU5viuqxZn97CfZhu5QrlvGvs,1838
@@ -719,7 +719,7 @@ zenml/utils/materializer_utils.py,sha256=xQlxAw3zqhKzfwYzx2_RyfFhym4amoVQfU1s28P
 zenml/utils/mlstacks_utils.py,sha256=Jv0ZRK8seBcnA96UelJhRaB6mRhZiA4062yE0yE7RaA,19949
 zenml/utils/networking_utils.py,sha256=zTDbOMkMPRWiWLZ2ccchd37rvvZWIIh0Kr9dZE-VJAs,7845
 zenml/utils/pagination_utils.py,sha256=nY6jg_RCQoeCipHrQfkdEarCah88P17IHMb4kswUhXY,1341
-zenml/utils/pipeline_docker_image_builder.py,sha256=I6gUYHacfyY1goo98p-NJFOHZ5GC55qSWmFAwzw819g,26346
+zenml/utils/pipeline_docker_image_builder.py,sha256=UT9bwVNdDUj3i6lIn0LFtjuxBViTsaKCPBck8V8-qSg,26886
 zenml/utils/proxy_utils.py,sha256=fgRlLa9pfXJDoxtB31_YP7DClOMQLek_nMmM0et6i3w,7241
 zenml/utils/pydantic_utils.py,sha256=pDpv_ciIj7o5RIgwWa0W6BUjmHjltcjWK4RRSoGVR2A,8468
 zenml/utils/secret_utils.py,sha256=0b0FeMhr28oxe4t6lOhnmHjENFADACynUEKeu6gBSkE,4019
@@ -1286,8 +1286,8 @@ zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=0CgWJo
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Cips_SPX7CtfYQyOZNiHAhmVtVgb17HCyHQia5MlTX4,8653
 zenml/zen_stores/sql_zen_store.py,sha256=f5spF_vWJg8kO6ud9uzELqh7LRxSAWyB1HLono1_gT0,353540
 zenml/zen_stores/zen_store_interface.py,sha256=vi0tr6vDES97h_rFZZYeaz2Y0nM4xm66CcUXSOEITN4,86372
-zenml_nightly-0.58.0.dev20240529.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.58.0.dev20240529.dist-info/METADATA,sha256=tTcDOsuledLpEXruCXbLQdYJkIHVoqB_nMq1RryAFZ4,19070
-zenml_nightly-0.58.0.dev20240529.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zenml_nightly-0.58.0.dev20240529.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.58.0.dev20240529.dist-info/RECORD,,
+zenml_nightly-0.58.0.dev20240531.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.58.0.dev20240531.dist-info/METADATA,sha256=33GkyT_LBxs5OkXANpeUp5cNc6_LELeEYNkys8Jau5w,19070
+zenml_nightly-0.58.0.dev20240531.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zenml_nightly-0.58.0.dev20240531.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.58.0.dev20240531.dist-info/RECORD,,