zenml-nightly 0.75.0.dev20250316__py3-none-any.whl → 0.75.0.dev20250318__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.75.0.dev20250316
+0.75.0.dev20250318

zenml/integrations/gcp/orchestrators/vertex_orchestrator.py

@@ -476,6 +476,11 @@ class VertexOrchestrator(ContainerizedOrchestrator, GoogleCredentialsMixin):
                             "Volume mounts are set but not supported in "
                             "Vertex with Kubeflow Pipelines 2.x. Ignoring..."
                         )
+                    if pod_settings.env or pod_settings.env_from:
+                        logger.warning(
+                            "Environment variables are set but not supported "
+                            "in Vertex with Vertex Pipelines 2.x. Ignoring..."
+                        )
                     for key in pod_settings.node_selectors:
                         if (
                             key

zenml/integrations/github/code_repositories/github_code_repository.py

@@ -20,6 +20,7 @@ from uuid import uuid4
 
 import requests
 from github import Consts, Github, GithubException
+from github.Auth import Token
 from github.Repository import Repository
 
 from zenml.code_repositories import (
@@ -145,9 +146,10 @@ class GitHubCodeRepository(BaseCodeRepository):
         """
         try:
             self._github_session = Github(
-                login_or_token=self.config.token,
                 base_url=self.config.api_url or Consts.DEFAULT_BASE_URL,
+                auth=Token(self.config.token) if self.config.token else None,
             )
+
             if self.config.token:
                 user = self._github_session.get_user().login
                 logger.debug(f"Logged in as {user}")

zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py

@@ -578,6 +578,12 @@ class KubeflowOrchestrator(ContainerizedOrchestrator):
                             "Volume mounts are set but not supported in "
                             "Kubeflow with Kubeflow Pipelines 2.x. Ignoring..."
                         )
+                    if pod_settings.env or pod_settings.env_from:
+                        logger.warning(
+                            "Environment variables are set but not supported "
+                            "in Kubeflow with Kubeflow Pipelines 2.x. "
+                            "Ignoring..."
+                        )
 
                     # apply pod settings
                     if (

zenml/integrations/kubernetes/orchestrators/manifest_utils.py

@@ -222,6 +222,18 @@ def add_pod_settings(
             else:
                 container.volume_mounts = settings.volume_mounts
 
+        if settings.env:
+            if container.env:
+                container.env.extend(settings.env)
+            else:
+                container.env = settings.env
+
+        if settings.env_from:
+            if container.env_from:
+                container.env_from.extend(settings.env_from)
+            else:
+                container.env_from = settings.env_from
+
     if settings.volumes:
         if pod_spec.volumes:
             pod_spec.volumes.extend(settings.volumes)

zenml/integrations/kubernetes/pod_settings.py

@@ -35,6 +35,8 @@ class KubernetesPodSettings(BaseSettings):
         host_ipc: Whether to enable host IPC for the pod.
         image_pull_secrets: Image pull secrets to use for the pod.
         labels: Labels to apply to the pod.
+        env: Environment variables to apply to the container.
+        env_from: Environment variables to apply to the container.
     """
 
     node_selectors: Dict[str, str] = {}
@@ -47,6 +49,8 @@ class KubernetesPodSettings(BaseSettings):
     host_ipc: bool = False
     image_pull_secrets: List[str] = []
     labels: Dict[str, str] = {}
+    env: List[Dict[str, Any]] = []
+    env_from: List[Dict[str, Any]] = []
 
     @field_validator("volumes", mode="before")
     @classmethod
@@ -155,3 +159,51 @@ class KubernetesPodSettings(BaseSettings):
             return serialization_utils.serialize_kubernetes_model(value)
         else:
             return value
+
+    @field_validator("env", mode="before")
+    @classmethod
+    def _convert_env(cls, value: Any) -> Any:
+        """Converts Kubernetes EnvVar to a dict.
+
+        Args:
+            value: The env value.
+
+        Returns:
+            The converted value.
+        """
+        from kubernetes.client.models import V1EnvVar
+
+        result = []
+        for element in value:
+            if isinstance(element, V1EnvVar):
+                result.append(
+                    serialization_utils.serialize_kubernetes_model(element)
+                )
+            else:
+                result.append(element)
+
+        return result
+
+    @field_validator("env_from", mode="before")
+    @classmethod
+    def _convert_env_from(cls, value: Any) -> Any:
+        """Converts Kubernetes EnvFromSource to a dict.
+
+        Args:
+            value: The env from value.
+
+        Returns:
+            The converted value.
+        """
+        from kubernetes.client.models import V1EnvFromSource
+
+        result = []
+        for element in value:
+            if isinstance(element, V1EnvFromSource):
+                result.append(
+                    serialization_utils.serialize_kubernetes_model(element)
+                )
+            else:
+                result.append(element)
+
+        return result

zenml/integrations/tekton/orchestrators/tekton_orchestrator.py

@@ -547,6 +547,11 @@ class TektonOrchestrator(ContainerizedOrchestrator):
                             "Volume mounts are set but not supported in "
                             "Tekton with Tekton Pipelines 2.x. Ignoring..."
                         )
+                    if pod_settings.env or pod_settings.env_from:
+                        logger.warning(
+                            "Environment variables are set but not supported "
+                            "in Tekton with Tekton Pipelines 2.x. Ignoring..."
+                        )
                     # apply pod settings
                     if (
                         KFP_ACCELERATOR_NODE_SELECTOR_CONSTRAINT_LABEL

zenml/login/pro/utils.py

@@ -35,6 +35,7 @@ def get_troubleshooting_instructions(url: str) -> str:
     credentials_store = get_credentials_store()
 
     credentials = credentials_store.get_credentials(url)
+    pro_api_url = None
     if credentials and credentials.type == ServerType.PRO:
         pro_api_url = credentials.pro_api_url or ZENML_PRO_API_URL
 

{zenml_nightly-0.75.0.dev20250316.dist-info → zenml_nightly-0.75.0.dev20250318.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.75.0.dev20250316
+Version: 0.75.0.dev20250318
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -79,12 +79,11 @@ Requires-Dist: kfp (>=2.6.0) ; extra == "vertex"
 Requires-Dist: kubernetes (>=18.20.0) ; extra == "connectors-kubernetes" or extra == "connectors-aws" or extra == "connectors-gcp" or extra == "connectors-azure"
 Requires-Dist: maison (<2.0) ; extra == "dev"
 Requires-Dist: mike (>=1.1.2,<2.0.0) ; extra == "dev"
-Requires-Dist: mkdocs (>=1.4.0,<2.0.0) ; extra == "dev"
-Requires-Dist: mkdocs-autorefs (>=0.4.1,<1.0.0) ; extra == "dev"
-Requires-Dist: mkdocs-awesome-pages-plugin (>=2.6.1,<3.0.0) ; extra == "dev"
-Requires-Dist: mkdocs-material (>=9.0.0,<10.0.0) ; extra == "dev"
-Requires-Dist: mkdocstrings (>=0.20.0,<1.0.0) ; extra == "dev"
-Requires-Dist: mkdocstrings-python (>=1.1.0,<2.0.0) ; extra == "dev"
+Requires-Dist: mkdocs (>=1.6.1,<2.0.0) ; extra == "dev"
+Requires-Dist: mkdocs-autorefs (>=1.4.0,<2.0.0) ; extra == "dev"
+Requires-Dist: mkdocs-awesome-pages-plugin (>=2.10.1,<3.0.0) ; extra == "dev"
+Requires-Dist: mkdocs-material (>=9.6.5,<10.0.0) ; extra == "dev"
+Requires-Dist: mkdocstrings[python] (>=0.28.1,<0.29.0) ; extra == "dev"
 Requires-Dist: mypy (==1.7.1) ; extra == "dev"
 Requires-Dist: orjson (>=3.10.0,<3.11.0) ; extra == "server"
 Requires-Dist: packaging (>=24.1)

{zenml_nightly-0.75.0.dev20250316.dist-info → zenml_nightly-0.75.0.dev20250318.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=SRamYg8RObx01aikckh_j2rSYXXhh4NnZvuAdSjY5qE,19
+zenml/VERSION,sha256=45FN9ROkAa_Ug1UpjB1qzVHjXhBemobOUKG4fNFxj0A,19
 zenml/__init__.py,sha256=MJlPgyX_3UnXfwAcxeFyJNxBKjmXjBoMHEWFh3hGXJc,2207
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -269,7 +269,7 @@ zenml/integrations/gcp/google_credentials_mixin.py,sha256=bPy3JYCCcyuTmPiVFqbY81
 zenml/integrations/gcp/image_builders/__init__.py,sha256=2IvTL6U2YpUoxGQXeXew-6WFoL5hHIxkqr4DaA5Ez9w,786
 zenml/integrations/gcp/image_builders/gcp_image_builder.py,sha256=5T6BXsHxLhvp1BF_rslXl1oZzykJUPuZ3E_7-9ZZYLk,9019
 zenml/integrations/gcp/orchestrators/__init__.py,sha256=6xLFJKZKQk73fHPF-XdpbQO87zjQNGTsNHjJjLfG_Kg,805
-zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=nv2dWvSWWW0TX-YHYX24udi3C9_ofkKKFc9DwXrKw2o,38922
+zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=TSCl_q-T9FHCh7pPUmkb3GgsLrTjvaV5mShdVcxd0Vw,39212
 zenml/integrations/gcp/service_connectors/__init__.py,sha256=fdydawaor8KAtMYvRZieiTuA1i5QATxXXgI-yV1lsn8,788
 zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=JGrFTKkQV4ZDn_yTEqX-AtBMraFasFgzLVws2mvhS64,94915
 zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_ojhoovL6SprmTx37It4EJ8,808
@@ -277,7 +277,7 @@ zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=X8CCniyAo7N
 zenml/integrations/gcp/vertex_custom_job_parameters.py,sha256=cK1Xf2sYoES_vRdMUbpnyrwzA9MTjaVp-p6CfKaY9rg,2475
 zenml/integrations/github/__init__.py,sha256=A8Yd--BbAG3HEfbWYOIEy_kzyLs2tBiawiLMosXd1Do,1467
 zenml/integrations/github/code_repositories/__init__.py,sha256=ub_hSE2ks2mZB1aeHRjQYz7QIRQIgOw2s080IIqJaGs,817
-zenml/integrations/github/code_repositories/github_code_repository.py,sha256=CBXrnLRu8wFJObBr9nTo2VU-4ObSNgf2Lg0AMZ3oewk,8252
+zenml/integrations/github/code_repositories/github_code_repository.py,sha256=VAFbiqzpnf4MDbXllGfc8bdYRFJ-7WAKQoYWGwnZl_o,8311
 zenml/integrations/github/plugins/__init__.py,sha256=yf7xkBs8wEUMP2-nFbDIVeXs1omHtZoyZBgobMYB1l0,804
 zenml/integrations/github/plugins/event_sources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/integrations/github/plugins/event_sources/github_webhook_event_source.py,sha256=nU--Wk9s2W7koF-JrtWIrSpuNWNVouftHQDiEY8dkdQ,17101
@@ -330,7 +330,7 @@ zenml/integrations/kubeflow/__init__.py,sha256=ab3v3sJBwo09vbyI2UQ5SnKfL0C9j8hrx
 zenml/integrations/kubeflow/flavors/__init__.py,sha256=l560A0oIpYeTpVVrdrVVYj-2-Y5CeyCSQMfwErZROxY,878
 zenml/integrations/kubeflow/flavors/kubeflow_orchestrator_flavor.py,sha256=b7W4oASHLYN9o2n__-W3zajb_MXhPaY0UutecsW0Xw4,10381
 zenml/integrations/kubeflow/orchestrators/__init__.py,sha256=J879DBt9WbpojBTdOXy7CO6F5OuTMntBeZ8aUY2EGO8,821
-zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py,sha256=H4rpm3f8k2MOeA4JZOByzkDHDCm4560EQKwnMU3Hnvg,39202
+zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py,sha256=tyJkjYYDvWl7FJJLM2Xh4E54QNj7aejGdN4Zn97u1UQ,39527
 zenml/integrations/kubeflow/orchestrators/local_deployment_utils.py,sha256=qszoOdvBpgIp40XkncphXAr9dRKnyZzGiz2mJ56bYmw,15448
 zenml/integrations/kubernetes/__init__.py,sha256=UzU5CaogX6ud5ChCK8JSZ06eoW18eIudbgntgPijYSc,1853
 zenml/integrations/kubernetes/flavors/__init__.py,sha256=a5gU45qCj3FkLwl_uVjlIkL_2F5DHk-w1gdcZrvVjBI,1266
@@ -341,8 +341,8 @@ zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=PIR6EICNaoxONtn
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=W6YpTQXXzQcpfskJcuYwN1LMiN8eUJQHWJ4QXMmADFs,22899
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=4BBy8-0xJahRjpkFc9fYy90uIRxOoy-s-GmXZeFE2tQ,6442
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=Y7uGU8eksMluGyXYsf688CwpiXwI_W6WYLscYwRZXRY,2494
-zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=urcPGUm51vRA1a5eZk2jEGjFrDLPHqD4jTVJDpxngnU,11486
-zenml/integrations/kubernetes/pod_settings.py,sha256=NMp4aHKRG29mh1Nq5uvV78Hzj1cMZj93poWCBiwov-M,4898
+zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=EfgsDYchOkpHPaZqSNbE_gF8a-3wQhDvSB1BCv7FW8A,11856
+zenml/integrations/kubernetes/pod_settings.py,sha256=fP2NeHf1XxT__D4g_6oHQf8pkiiamFUFYmfNor2OoP8,6386
 zenml/integrations/kubernetes/serialization_utils.py,sha256=cPSe4szdBLzDnUZT9nQc2CCA8h84aj5oTA8vsUE36ig,7000
 zenml/integrations/kubernetes/service_connectors/__init__.py,sha256=Uf6zlHIapYrRDl3xOPWQ2jA7jt85SXx1U7DmSxzxTvQ,818
 zenml/integrations/kubernetes/service_connectors/kubernetes_service_connector.py,sha256=Cv4tiVxoQOz9ex0lf3JdJrooEkgMwfDfwt5GOeNRpQU,19669
@@ -531,7 +531,7 @@ zenml/integrations/tekton/__init__.py,sha256=WukQ041QAzuFIy5xXAyLRBFE-61fQqyTfX-
 zenml/integrations/tekton/flavors/__init__.py,sha256=-S5XuwcZKWyGb9tVfl_gEFJj1KSXX6kwIfe-EhE0TmM,864
 zenml/integrations/tekton/flavors/tekton_orchestrator_flavor.py,sha256=XgFgzJUlje9J1o5zwBvs_ycpgQjGdi50KZFA9_tT0vc,8268
 zenml/integrations/tekton/orchestrators/__init__.py,sha256=yCJEM-PCechO4DF_YQeg82y76nwBKeXTy_SIXRWX2DE,811
-zenml/integrations/tekton/orchestrators/tekton_orchestrator.py,sha256=t0TsocPqGMb92rSKKujhapU34jlJ2LO7HCghZVeOkbM,35282
+zenml/integrations/tekton/orchestrators/tekton_orchestrator.py,sha256=aL0v_vYuwiAW3OKzXDdB_qdOqBFS2pC0sjKl2xewGVw,35572
 zenml/integrations/tensorboard/__init__.py,sha256=68Js570rS5PwoWMkrjGtadGg3j5JeGbYVczfpDTnL3g,1538
 zenml/integrations/tensorboard/services/__init__.py,sha256=Y-YVPLxJkFSpXNDZSk8gEBISsXJGk_DgGIyVIIFIa40,798
 zenml/integrations/tensorboard/services/tensorboard_service.py,sha256=nSNWeUDMM4H1hjFL8FireSxQI5SeLl75GJVmY83yXAg,4398
@@ -588,7 +588,7 @@ zenml/login/pro/models.py,sha256=UyIozykWVyq7SV1yk5e81YBcT7rAfkPmm226dVeSLNY,907
 zenml/login/pro/organization/__init__.py,sha256=zK8RdBK_vgL_pEJyUlbpjf7m61EQ12-C-Bm7gN4F2D8,650
 zenml/login/pro/organization/client.py,sha256=xcYfj2inTj_lTDIMwAFgMyZNo0QV5T55IAYlXUwyhKk,2148
 zenml/login/pro/organization/models.py,sha256=w3eNH3PCOEvpuy4F58OTjubtUbHCrcSIX_r_le2Xru8,983
-zenml/login/pro/utils.py,sha256=iI9wovi7eGsRf_O44hywhPkAPTR_RwhtPX1GiPrS9ts,5175
+zenml/login/pro/utils.py,sha256=6JWo1mxMwedhM8mETFKb-8UlUCjWttRnvSEDKmwQIS4,5198
 zenml/login/pro/workspace/__init__.py,sha256=WHKwRfiYsFD3HXMOqWiwMndwWEqqt5Ssi37uj73cmlw,647
 zenml/login/pro/workspace/client.py,sha256=S5kasWuEd8MMFkgAxbjvzrmYGQr4-evwp2Aw45Ng6FI,2824
 zenml/login/pro/workspace/models.py,sha256=godIY1q2dUxB0QILaOFOeqV1I3WglWUjZC11nDWVxSw,5408
@@ -1306,8 +1306,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=rdDJBFWNGuURDGY58701BzMWeWIIwqzrKl4QNZ6uOXU,440274
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.75.0.dev20250316.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.75.0.dev20250316.dist-info/METADATA,sha256=9ETnOcC83wOMwVXGVIK6-a-M-3nul2lDZjRMMfGHdj4,24274
-zenml_nightly-0.75.0.dev20250316.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-zenml_nightly-0.75.0.dev20250316.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.75.0.dev20250316.dist-info/RECORD,,
+zenml_nightly-0.75.0.dev20250318.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.75.0.dev20250318.dist-info/METADATA,sha256=VaDygNA-1zBgmMR9YxUytWNTCgMrSwGBnkig2O2Z6bA,24215
+zenml_nightly-0.75.0.dev20250318.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+zenml_nightly-0.75.0.dev20250318.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.75.0.dev20250318.dist-info/RECORD,,