zenml-nightly 0.73.0.dev20250127__py3-none-any.whl → 0.73.0.dev20250128__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.73.0.dev20250127
+0.73.0.dev20250128

zenml/cli/code_repository.py

@@ -42,7 +42,7 @@ def code_repository() -> None:
     context_settings={"ignore_unknown_options": True},
     help="Register a code repository.",
 )
-@click.argument("name", type=click.STRING)
+@click.argument("name", type=str)
 @click.option(
     "--type",
     "-t",
@@ -183,6 +183,74 @@ def list_code_repositories(**kwargs: Any) -> None:
         )
 
 
+@code_repository.command(
+    "update",
+    help="Update a code repository.",
+    context_settings={"ignore_unknown_options": True},
+)
+@click.argument("name_or_id", type=str, required=True)
+@click.option(
+    "--name",
+    "-n",
+    type=str,
+    required=False,
+    help="The new code repository name.",
+)
+@click.option(
+    "--description",
+    "-d",
+    type=str,
+    required=False,
+    help="The new code repository description.",
+)
+@click.option(
+    "--logo-url",
+    "-l",
+    type=str,
+    required=False,
+    help="New URL of a logo (png, jpg or svg) for the code repository.",
+)
+@click.argument(
+    "args",
+    nargs=-1,
+    type=click.UNPROCESSED,
+)
+def update_code_repository(
+    name_or_id: str,
+    name: Optional[str],
+    description: Optional[str],
+    logo_url: Optional[str],
+    args: List[str],
+) -> None:
+    """Update a code repository.
+
+    Args:
+        name_or_id: Name or ID of the code repository to update.
+        name: New name of the code repository.
+        description: New description of the code repository.
+        logo_url: New logo URL of the code repository.
+        args: Code repository configurations.
+    """
+    parsed_name_or_id, parsed_args = cli_utils.parse_name_and_extra_arguments(
+        list(args) + [name_or_id], expand_args=True, name_mandatory=True
+    )
+    assert parsed_name_or_id
+
+    with console.status(
+        f"Updating code repository '{parsed_name_or_id}'...\n"
+    ):
+        Client().update_code_repository(
+            name_id_or_prefix=parsed_name_or_id,
+            name=name,
+            description=description,
+            logo_url=logo_url,
+            config=parsed_args,
+        )
+        cli_utils.declare(
+            f"Successfully updated code repository `{parsed_name_or_id}`."
+        )
+
+
 @code_repository.command("delete")
 @click.argument("name_or_id", type=str, required=True)
 @click.option(

zenml/client.py

@@ -4952,25 +4952,15 @@ class Client(metaclass=ClientMetaClass):
 
     # --------------------------- Code repositories ---------------------------
 
-    def create_code_repository(
-        self,
-        name: str,
-        config: Dict[str, Any],
-        source: Source,
-        description: Optional[str] = None,
-        logo_url: Optional[str] = None,
-    ) -> CodeRepositoryResponse:
-        """Create a new code repository.
+    @staticmethod
+    def _validate_code_repository_config(
+        source: Source, config: Dict[str, Any]
+    ) -> None:
+        """Validate a code repository config.
 
         Args:
-            name: Name of the code repository.
-            config: The configuration for the code repository.
-            source: The code repository implementation source.
-            description: The code repository description.
-            logo_url: URL of a logo (png, jpg or svg) for the code repository.
-
-        Returns:
-            The created code repository.
+            source: The code repository source.
+            config: The code repository config.
 
         Raises:
             RuntimeError: If the provided config is invalid.
@@ -4983,13 +4973,38 @@ class Client(metaclass=ClientMetaClass):
             )
         )
         try:
-            # Validate the repo config
+            # This does a login to verify the credentials
             code_repo_class(id=uuid4(), config=config)
+
+            # Explicitly access the config for pydantic validation, in case
+            # the login for some reason did not do that.
+            _ = code_repo_class.config
         except Exception as e:
             raise RuntimeError(
                 "Failed to validate code repository config."
             ) from e
 
+    def create_code_repository(
+        self,
+        name: str,
+        config: Dict[str, Any],
+        source: Source,
+        description: Optional[str] = None,
+        logo_url: Optional[str] = None,
+    ) -> CodeRepositoryResponse:
+        """Create a new code repository.
+
+        Args:
+            name: Name of the code repository.
+            config: The configuration for the code repository.
+            source: The code repository implementation source.
+            description: The code repository description.
+            logo_url: URL of a logo (png, jpg or svg) for the code repository.
+
+        Returns:
+            The created code repository.
+        """
+        self._validate_code_repository_config(source=source, config=config)
         repo_request = CodeRepositoryRequest(
             user=self.active_user.id,
             workspace=self.active_workspace.id,
@@ -5088,6 +5103,7 @@ class Client(metaclass=ClientMetaClass):
         name: Optional[str] = None,
         description: Optional[str] = None,
         logo_url: Optional[str] = None,
+        config: Optional[Dict[str, Any]] = None,
     ) -> CodeRepositoryResponse:
         """Update a code repository.
 
@@ -5097,6 +5113,10 @@ class Client(metaclass=ClientMetaClass):
             name: New name of the code repository.
             description: New description of the code repository.
             logo_url: New logo URL of the code repository.
+            config: New configuration options for the code repository. Will
+                be used to update the existing configuration values. To remove
+                values from the existing configuration, set the value for that
+                key to `None`.
 
         Returns:
             The updated code repository.
@@ -5107,6 +5127,18 @@ class Client(metaclass=ClientMetaClass):
         update = CodeRepositoryUpdate(
             name=name, description=description, logo_url=logo_url
         )
+        if config is not None:
+            combined_config = repo.config
+            combined_config.update(config)
+            combined_config = {
+                k: v for k, v in combined_config.items() if v is not None
+            }
+
+            self._validate_code_repository_config(
+                source=repo.source, config=combined_config
+            )
+            update.config = combined_config
+
         return self.zen_store.update_code_repository(
             code_repository_id=repo.id, update=update
         )

zenml/config/schedule.py

@@ -13,10 +13,15 @@
 #  permissions and limitations under the License.
 """Class for defining a pipeline schedule."""
 
-import datetime
+from datetime import datetime, timedelta
 from typing import Optional
 
-from pydantic import BaseModel, model_validator
+from pydantic import (
+    BaseModel,
+    ValidationInfo,
+    field_validator,
+    model_validator,
+)
 
 from zenml.logger import get_logger
 
@@ -32,8 +37,12 @@ class Schedule(BaseModel):
             and time.
         cron_expression: Cron expression for the pipeline schedule. If a value
             for this is set it takes precedence over the start time + interval.
-        start_time: datetime object to indicate when to start the schedule.
-        end_time: datetime object to indicate when to end the schedule.
+        start_time: When the schedule should start. If this is a datetime object
+            without any timezone, it is treated as a datetime in the local
+            timezone.
+        end_time: When the schedule should end. If this is a datetime object
+            without any timezone, it is treated as a datetime in the local
+            timezone.
         interval_second: datetime timedelta indicating the seconds between two
             recurring runs for a periodic schedule.
         catchup: Whether the recurring run should catch up if behind schedule.
@@ -43,17 +52,45 @@ class Schedule(BaseModel):
             schedules the latest interval if more than one interval is ready to
             be scheduled. Usually, if your pipeline handles backfill
             internally, you should turn catchup off to avoid duplicate backfill.
-        run_once_start_time: datetime object to indicate when to run the
-            pipeline once. This is useful for one-off runs.
+        run_once_start_time: When to run the pipeline once. If this is a
+            datetime object without any timezone, it is treated as a datetime
+            in the local timezone.
     """
 
     name: Optional[str] = None
     cron_expression: Optional[str] = None
-    start_time: Optional[datetime.datetime] = None
-    end_time: Optional[datetime.datetime] = None
-    interval_second: Optional[datetime.timedelta] = None
+    start_time: Optional[datetime] = None
+    end_time: Optional[datetime] = None
+    interval_second: Optional[timedelta] = None
     catchup: bool = False
-    run_once_start_time: Optional[datetime.datetime] = None
+    run_once_start_time: Optional[datetime] = None
+
+    @field_validator(
+        "start_time", "end_time", "run_once_start_time", mode="after"
+    )
+    @classmethod
+    def _ensure_timezone(
+        cls, value: Optional[datetime], info: ValidationInfo
+    ) -> Optional[datetime]:
+        """Ensures that all datetimes are timezone aware.
+
+        Args:
+            value: The datetime.
+            info: The validation info.
+
+        Returns:
+            A timezone aware datetime or None.
+        """
+        if value and value.tzinfo is None:
+            assert info.field_name
+            logger.warning(
+                "Your schedule `%s` is missing a timezone. It will be treated "
+                "as a datetime in your local timezone.",
+                info.field_name,
+            )
+            value = value.astimezone()
+
+        return value
 
     @model_validator(mode="after")
     def _ensure_cron_or_periodic_schedule_configured(self) -> "Schedule":

zenml/entrypoints/base_entrypoint_configuration.py

@@ -214,12 +214,14 @@ class BaseEntrypointConfiguration(ABC):
         if not should_download_code:
             return
 
-        if code_reference := deployment.code_reference:
+        if code_path := deployment.code_path:
+            code_utils.download_code_from_artifact_store(code_path=code_path)
+        elif code_reference := deployment.code_reference:
+            # TODO: This might fail if the code repository had unpushed changes
+            # at the time the pipeline run was started.
             self.download_code_from_code_repository(
                 code_reference=code_reference
             )
-        elif code_path := deployment.code_path:
-            code_utils.download_code_from_artifact_store(code_path=code_path)
         else:
             raise RuntimeError(
                 "Code download required but no code reference or path provided."

zenml/integrations/github/code_repositories/github_code_repository.py

@@ -18,7 +18,7 @@ import re
 from typing import List, Optional
 
 import requests
-from github import Github, GithubException
+from github import Consts, Github, GithubException
 from github.Repository import Repository
 
 from zenml.code_repositories import (
@@ -30,6 +30,7 @@ from zenml.code_repositories.base_code_repository import (
 )
 from zenml.code_repositories.git import LocalGitRepositoryContext
 from zenml.logger import get_logger
+from zenml.utils import deprecation_utils
 from zenml.utils.secret_utils import SecretField
 
 logger = get_logger(__name__)
@@ -39,19 +40,24 @@ class GitHubCodeRepositoryConfig(BaseCodeRepositoryConfig):
     """Config for GitHub code repositories.
 
     Args:
-        url: The URL of the GitHub instance.
+        api_url: The GitHub API URL.
         owner: The owner of the repository.
         repository: The name of the repository.
         host: The host of the repository.
         token: The token to access the repository.
     """
 
-    url: Optional[str]
+    api_url: Optional[str] = None
     owner: str
     repository: str
     host: Optional[str] = "github.com"
     token: Optional[str] = SecretField(default=None)
 
+    url: Optional[str] = None
+    _deprecation_validator = deprecation_utils.deprecate_pydantic_attributes(
+        ("url", "api_url")
+    )
+
 
 class GitHubCodeRepository(BaseCodeRepository):
     """GitHub code repository."""
@@ -95,7 +101,7 @@ class GitHubCodeRepository(BaseCodeRepository):
         Raises:
             RuntimeError: If the repository is not public.
         """
-        url = f"https://api.github.com/repos/{owner}/{repo}"
+        url = f"{Consts.DEFAULT_BASE_URL}/repos/{owner}/{repo}"
         response = requests.get(url, timeout=7)
 
         try:
@@ -103,12 +109,15 @@ class GitHubCodeRepository(BaseCodeRepository):
                 pass
             else:
                 raise RuntimeError(
-                    "It is not possible to access this repository as it does not appear to be public."
-                    "Access to private repositories is only possible when a token is provided. Please provide a token and try again"
+                    "It is not possible to access this repository as it does "
+                    "not appear to be public. Access to private repositories "
+                    "is only possible when a token is provided. Please provide "
+                    "a token and try again"
                 )
         except Exception as e:
             raise RuntimeError(
-                f"An error occurred while checking if repository is public: {str(e)}"
+                "An error occurred while checking if repository is public: "
+                f"{str(e)}"
             )
 
     def login(
@@ -120,7 +129,10 @@ class GitHubCodeRepository(BaseCodeRepository):
             RuntimeError: If the login fails.
         """
         try:
-            self._github_session = Github(self.config.token)
+            self._github_session = Github(
+                login_or_token=self.config.token,
+                base_url=self.config.api_url or Consts.DEFAULT_BASE_URL,
+            )
             if self.config.token:
                 user = self._github_session.get_user().login
                 logger.debug(f"Logged in as {user}")

zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py

@@ -31,6 +31,7 @@ from zenml.code_repositories.git.local_git_repository_context import (
     LocalGitRepositoryContext,
 )
 from zenml.logger import get_logger
+from zenml.utils import deprecation_utils
 from zenml.utils.secret_utils import SecretField
 
 logger = get_logger(__name__)
@@ -40,19 +41,24 @@ class GitLabCodeRepositoryConfig(BaseCodeRepositoryConfig):
     """Config for GitLab code repositories.
 
     Args:
-        url: The full URL of the GitLab project.
+        instance_url: The URL of the GitLab instance.
         group: The group of the project.
         project: The name of the GitLab project.
         host: The host of GitLab in case it is self-hosted instance.
         token: The token to access the repository.
     """
 
-    url: Optional[str]
+    instance_url: Optional[str] = None
     group: str
     project: str
     host: Optional[str] = "gitlab.com"
     token: str = SecretField()
 
+    url: Optional[str] = None
+    _deprecation_validator = deprecation_utils.deprecate_pydantic_attributes(
+        ("url", "instance_url")
+    )
+
 
 class GitLabCodeRepository(BaseCodeRepository):
     """GitLab code repository."""
@@ -85,7 +91,7 @@ class GitLabCodeRepository(BaseCodeRepository):
         """
         try:
             self._gitlab_session = Gitlab(
-                self.config.url, private_token=self.config.token
+                url=self.config.instance_url, private_token=self.config.token
             )
             self._gitlab_session.auth()
             user = self._gitlab_session.user or None

zenml/models/v2/core/schedule.py

@@ -13,11 +13,11 @@
 #  permissions and limitations under the License.
 """Models representing schedules."""
 
-import datetime
+from datetime import datetime, timedelta, timezone
 from typing import Dict, Optional, Union
 from uuid import UUID
 
-from pydantic import Field, model_validator
+from pydantic import Field, field_validator, model_validator
 
 from zenml.constants import STR_FIELD_MAX_LENGTH
 from zenml.logger import get_logger
@@ -46,15 +46,36 @@ class ScheduleRequest(WorkspaceScopedRequest):
     active: bool
 
     cron_expression: Optional[str] = None
-    start_time: Optional[datetime.datetime] = None
-    end_time: Optional[datetime.datetime] = None
-    interval_second: Optional[datetime.timedelta] = None
+    start_time: Optional[datetime] = None
+    end_time: Optional[datetime] = None
+    interval_second: Optional[timedelta] = None
     catchup: bool = False
-    run_once_start_time: Optional[datetime.datetime] = None
+    run_once_start_time: Optional[datetime] = None
 
     orchestrator_id: Optional[UUID]
     pipeline_id: Optional[UUID]
 
+    @field_validator(
+        "start_time", "end_time", "run_once_start_time", mode="after"
+    )
+    @classmethod
+    def _ensure_tzunaware_utc(
+        cls, value: Optional[datetime]
+    ) -> Optional[datetime]:
+        """Ensures that all datetimes are timezone unaware and in UTC time.
+
+        Args:
+            value: The datetime.
+
+        Returns:
+            The datetime in UTC time without timezone.
+        """
+        if value and value.tzinfo:
+            value = value.astimezone(timezone.utc)
+            value = value.replace(tzinfo=None)
+
+        return value
+
     @model_validator(mode="after")
     def _ensure_cron_or_periodic_schedule_configured(
         self,
@@ -108,11 +129,11 @@ class ScheduleUpdate(BaseUpdate):
     name: Optional[str] = None
     active: Optional[bool] = None
     cron_expression: Optional[str] = None
-    start_time: Optional[datetime.datetime] = None
-    end_time: Optional[datetime.datetime] = None
-    interval_second: Optional[datetime.timedelta] = None
+    start_time: Optional[datetime] = None
+    end_time: Optional[datetime] = None
+    interval_second: Optional[timedelta] = None
     catchup: Optional[bool] = None
-    run_once_start_time: Optional[datetime.datetime] = None
+    run_once_start_time: Optional[datetime] = None
     orchestrator_id: Optional[UUID] = None
     pipeline_id: Optional[UUID] = None
 
@@ -125,11 +146,11 @@ class ScheduleResponseBody(WorkspaceScopedResponseBody):
 
     active: bool
     cron_expression: Optional[str] = None
-    start_time: Optional[datetime.datetime] = None
-    end_time: Optional[datetime.datetime] = None
-    interval_second: Optional[datetime.timedelta] = None
+    start_time: Optional[datetime] = None
+    end_time: Optional[datetime] = None
+    interval_second: Optional[timedelta] = None
     catchup: bool = False
-    run_once_start_time: Optional[datetime.datetime] = None
+    run_once_start_time: Optional[datetime] = None
 
 
 class ScheduleResponseMetadata(WorkspaceScopedResponseMetadata):
@@ -217,7 +238,7 @@ class ScheduleResponse(
         return self.get_body().cron_expression
 
     @property
-    def start_time(self) -> Optional[datetime.datetime]:
+    def start_time(self) -> Optional[datetime]:
         """The `start_time` property.
 
         Returns:
@@ -226,7 +247,7 @@ class ScheduleResponse(
         return self.get_body().start_time
 
     @property
-    def end_time(self) -> Optional[datetime.datetime]:
+    def end_time(self) -> Optional[datetime]:
         """The `end_time` property.
 
         Returns:
@@ -235,7 +256,7 @@ class ScheduleResponse(
         return self.get_body().end_time
 
     @property
-    def run_once_start_time(self) -> Optional[datetime.datetime]:
+    def run_once_start_time(self) -> Optional[datetime]:
         """The `run_once_start_time` property.
 
         Returns:
@@ -244,7 +265,7 @@ class ScheduleResponse(
         return self.get_body().run_once_start_time
 
     @property
-    def interval_second(self) -> Optional[datetime.timedelta]:
+    def interval_second(self) -> Optional[timedelta]:
         """The `interval_second` property.
 
         Returns:
@@ -313,10 +334,10 @@ class ScheduleFilter(WorkspaceScopedFilter):
         default=None,
         description="The cron expression, describing the schedule",
     )
-    start_time: Optional[Union[datetime.datetime, str]] = Field(
+    start_time: Optional[Union[datetime, str]] = Field(
         default=None, description="Start time", union_mode="left_to_right"
     )
-    end_time: Optional[Union[datetime.datetime, str]] = Field(
+    end_time: Optional[Union[datetime, str]] = Field(
         default=None, description="End time", union_mode="left_to_right"
     )
     interval_second: Optional[Optional[float]] = Field(
@@ -332,7 +353,7 @@ class ScheduleFilter(WorkspaceScopedFilter):
         default=None,
         description="Name of the schedule",
     )
-    run_once_start_time: Optional[Union[datetime.datetime, str]] = Field(
+    run_once_start_time: Optional[Union[datetime, str]] = Field(
         default=None,
         description="The time at which the schedule should run once",
         union_mode="left_to_right",

zenml/pipelines/build_utils.py

@@ -30,7 +30,6 @@ from zenml.code_repositories import BaseCodeRepository
 from zenml.logger import get_logger
 from zenml.models import (
     BuildItem,
-    CodeReferenceRequest,
     PipelineBuildBase,
     PipelineBuildRequest,
     PipelineBuildResponse,
@@ -362,6 +361,7 @@ def create_pipeline_build(
             item_key = checksums[checksum]
             image_name_or_digest = images[item_key].image
             contains_code = images[item_key].contains_code
+            requires_code_download = images[item_key].requires_code_download
             dockerfile = images[item_key].dockerfile
             requirements = images[item_key].requirements
         else:
@@ -373,7 +373,7 @@ def create_pipeline_build(
             include_files = build_config.should_include_files(
                 code_repository=code_repository,
             )
-            download_files = build_config.should_download_files(
+            requires_code_download = build_config.should_download_files(
                 code_repository=code_repository,
             )
             pass_code_repo = (
@@ -391,7 +391,6 @@ def create_pipeline_build(
                 tag=tag,
                 stack=stack,
                 include_files=include_files,
-                download_files=download_files,
                 entrypoint=build_config.entrypoint,
                 extra_files=build_config.extra_files,
                 code_repository=code_repository if pass_code_repo else None,
@@ -404,7 +403,7 @@ def create_pipeline_build(
             requirements=requirements,
             settings_checksum=checksum,
             contains_code=contains_code,
-            requires_code_download=download_files,
+            requires_code_download=requires_code_download,
         )
         checksums[checksum] = combined_key
 
@@ -537,6 +536,14 @@ def verify_local_repository_context(
         )
         code_repository = BaseCodeRepository.from_model(model)
 
+        if will_download_from_code_repository(
+            deployment=deployment, local_repo_context=local_repo_context
+        ):
+            logger.info(
+                "Using code repository `%s` to download code for this run.",
+                model.name,
+            )
+
     return code_repository
 
 
@@ -695,14 +702,15 @@ def compute_stack_checksum(stack: StackResponse) -> str:
 def should_upload_code(
     deployment: PipelineDeploymentBase,
     build: Optional[PipelineBuildResponse],
-    code_reference: Optional[CodeReferenceRequest],
+    can_download_from_code_repository: bool,
 ) -> bool:
     """Checks whether the current code should be uploaded for the deployment.
 
     Args:
         deployment: The deployment.
         build: The build for the deployment.
-        code_reference: The code reference for the deployment.
+        can_download_from_code_repository: Whether the code can be downloaded
+            from a code repository.
 
     Returns:
         Whether the current code should be uploaded for the deployment.
@@ -718,7 +726,7 @@ def should_upload_code(
         docker_settings = step.config.docker_settings
 
         if (
-            code_reference
+            can_download_from_code_repository
             and docker_settings.allow_download_from_code_repository
         ):
             # No upload needed for this step
@@ -728,3 +736,31 @@ def should_upload_code(
             return True
 
     return False
+
+
+def will_download_from_code_repository(
+    deployment: PipelineDeploymentBase,
+    local_repo_context: "LocalRepositoryContext",
+) -> bool:
+    """Checks whether a code repository will be used to download code.
+
+    Args:
+        deployment: The deployment.
+        local_repo_context: The local repository context.
+
+    Returns:
+        Whether a code repository will be used to download code.
+    """
+    if not build_required(deployment=deployment):
+        return False
+
+    if local_repo_context.has_local_changes:
+        return False
+
+    for step in deployment.step_configurations.values():
+        docker_settings = step.config.docker_settings
+
+        if docker_settings.allow_download_from_code_repository:
+            return True
+
+    return False

zenml/pipelines/pipeline_definition.py

@@ -701,6 +701,7 @@ To avoid this consider setting pipeline parameters only in one place (config or
         code_repository = build_utils.verify_local_repository_context(
             deployment=deployment, local_repo_context=local_repo_context
         )
+        can_download_from_code_repository = code_repository is not None
 
         if prevent_build_reuse:
             logger.warning(
@@ -737,7 +738,7 @@ To avoid this consider setting pipeline parameters only in one place (config or
         if build_utils.should_upload_code(
             deployment=deployment,
             build=build_model,
-            code_reference=code_reference,
+            can_download_from_code_repository=can_download_from_code_repository,
         ):
             code_archive = code_utils.CodeArchive(
                 root=source_utils.get_source_root()

zenml/utils/code_repository_utils.py

@@ -109,9 +109,16 @@ def find_active_code_repository(
     for model in depaginate(list_method=Client().list_code_repositories):
         try:
             repo = BaseCodeRepository.from_model(model)
-        except Exception:
+        except ImportError:
             logger.debug(
-                "Failed to instantiate code repository class.", exc_info=True
+                "Failed to import code repository class.", exc_info=True
+            )
+            continue
+        except Exception as e:
+            logger.warning(
+                "Failed to instantiate or login to code repository `%s`: %s",
+                model.name,
+                e,
             )
             continue
 

zenml/utils/deprecation_utils.py

@@ -132,14 +132,16 @@ def deprecate_pydantic_attributes(
             if replacement_attribute is None:
                 _warn(
                     message=f"The attribute `{deprecated_attribute}` of class "
-                    f"`{cls.__name__}` will be deprecated soon.",
+                    f"`{cls.__name__}` is deprecated and will be removed in "
+                    "the future.",
                     attribute=deprecated_attribute,
                 )
                 continue
 
             _warn(
                 message=f"The attribute `{deprecated_attribute}` of class "
-                f"`{cls.__name__}` will be deprecated soon. Use the "
+                f"`{cls.__name__}` is deprecated and will be removed in the "
+                "future. Use the "
                 f"attribute `{replacement_attribute}` instead.",
                 attribute=deprecated_attribute,
             )

zenml/utils/pipeline_docker_image_builder.py

@@ -78,7 +78,6 @@ class PipelineDockerImageBuilder:
         tag: str,
         stack: "Stack",
         include_files: bool,
-        download_files: bool,
         entrypoint: Optional[str] = None,
         extra_files: Optional[Dict[str, str]] = None,
         code_repository: Optional["BaseCodeRepository"] = None,
@@ -93,7 +92,6 @@ class PipelineDockerImageBuilder:
             tag: The tag to use for the image.
             stack: The stack on which the pipeline will be deployed.
             include_files: Whether to include files in the build context.
-            download_files: Whether to download files in the build context.
             entrypoint: Entrypoint to use for the final image. If left empty,
                 no entrypoint will be included in the image.
             extra_files: Extra files to add to the build context. Keys are the
@@ -165,7 +163,6 @@ class PipelineDockerImageBuilder:
                 docker_settings.apt_packages,
                 docker_settings.environment,
                 include_files,
-                download_files,
                 entrypoint,
                 extra_files,
             ]

zenml/zen_stores/schemas/code_repository_schemas.py

@@ -151,6 +151,9 @@ class CodeRepositorySchema(NamedSchema, table=True):
         if update.logo_url:
             self.logo_url = update.logo_url
 
+        if update.config:
+            self.config = json.dumps(update.config)
+
         self.updated = utc_now()
         return self
 

{zenml_nightly-0.73.0.dev20250127.dist-info → zenml_nightly-0.73.0.dev20250128.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.73.0.dev20250127
+Version: 0.73.0.dev20250128
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.73.0.dev20250127.dist-info → zenml_nightly-0.73.0.dev20250128.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=JkGnFvDvS5Ka0VljyXw-YlBu7Vxd-DgMGHbZytdGcEo,19
+zenml/VERSION,sha256=idxxoPzqv46tqwm3PHoXsGFsEa0kogFk5FNLMlg5-3Y,19
 zenml/__init__.py,sha256=SkMObQA41ajqdZqGErN00S1Vf3KAxpLvbZ-OBy5uYoo,2130
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -32,7 +32,7 @@ zenml/cli/artifact.py,sha256=7lsAS52DroBTFkFWxkyb-lIDOGP5jPL_Se_RDG_2jgg,9564
 zenml/cli/authorized_device.py,sha256=_1PzE3BM2SmwtuzRliEMStvbBRKWQmg_lbwCRtn8dBg,4324
 zenml/cli/base.py,sha256=CGWqHb5A8k4XwzOQ4cdV25zWb9qOZr-xOq46wL3we1M,28243
 zenml/cli/cli.py,sha256=Pnq468IZ4oqzluA_gZ5PsrdnSPEyHcasIH-xI1_8Y_Q,5454
-zenml/cli/code_repository.py,sha256=7DNJMc7RL8GaU8DwX0mDSViLH9oclqhsX2AU-VWOKb8,6979
+zenml/cli/code_repository.py,sha256=UEgiVTzb184AaSrNmO-ylaZ9f_OXXe76EHbf9zrZVc8,8746
 zenml/cli/config.py,sha256=UI_j0a_zRgEUd2q0zuOi4UgbjiCYjMJ_Y9iSg-wi8Oo,2768
 zenml/cli/downgrade.py,sha256=eTpXts8y4s3wEUwOlvZGWsTngoMV8Stuzj0K-SAQUGU,1887
 zenml/cli/feature.py,sha256=Q8tNvWBlRze3FUyn0_VpOdl316ZW87476j7ezJb16GA,4387
@@ -55,7 +55,7 @@ zenml/cli/user_management.py,sha256=fTuRworQahst_j78qPYTtgciUeUOxwo7efiyPwmj2tI,
 zenml/cli/utils.py,sha256=aCSQFjfaLtBt4OYzYDwO2Rmdmn-b5AVBqpBGvv7OT7E,86420
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
 zenml/cli/workspace.py,sha256=bp02aXou574ToWPD8OAIB_cg3mvpE011H8aMKegT-nU,2970
-zenml/client.py,sha256=OaPYruL6NKMFhp4zRERSpR1xPHKBjiLnDa71Ouxs4TE,282139
+zenml/client.py,sha256=f0ySssFW_m7G5cj9UKEeA0Tm-e8GwfjIvLaVuixHnKM,283418
 zenml/client_lazy_loader.py,sha256=MOBgS1ITYqGvPUnWQ6edn9s8Hr_72YfWbwEIfHKUr9g,7104
 zenml/code_repositories/__init__.py,sha256=W5bDfzAG8OXIKZSV1L-VHuzMcSCYa9qzTdPb3jqfyYw,920
 zenml/code_repositories/base_code_repository.py,sha256=_DbxIBxvJlN0PFhST0vlTIQ26Q6V3Nar0kYdeGaJrk8,4386
@@ -74,7 +74,7 @@ zenml/config/pipeline_run_configuration.py,sha256=RoX2lGjWGm_0aE2ZMfsREjo0QnwWb4
 zenml/config/pipeline_spec.py,sha256=uWpiIfznJvXAiKs1yMIUDT1h1tYEFNO-RDVTYcIv9CE,2821
 zenml/config/resource_settings.py,sha256=bl3xahx--XS9p1CsTDSuvkZX9Oxb-Zj85UpefR8WrYs,3899
 zenml/config/retry_config.py,sha256=4UH1xqw0G6fSEbXSNKfmiFEkwadxQef9BGMe3JBm6NI,929
-zenml/config/schedule.py,sha256=aXHoOn0nvUw0YPXIlzS7idosuW2DMMAmkRqVdOEQe2o,4244
+zenml/config/schedule.py,sha256=qtMWa-mEo7jIKvDzQUstMwe57gdbvyWAQ7ggsoddbCA,5349
 zenml/config/secret_reference_mixin.py,sha256=6DOgOH_w1_coNs1NasMjY8SOv0MqzWQZxwrWsfncsgs,5862
 zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wzrx-zHTU,2818
 zenml/config/server_config.py,sha256=SR0JerBn-NCQRflQ0WuDWEsjM4wc9VBoqwx4CF59Lpc,31192
@@ -96,7 +96,7 @@ zenml/container_registries/github_container_registry.py,sha256=rbcGkFfPDk-Il0T9F
 zenml/data_validators/__init__.py,sha256=9Fa0jiUSQ_JsLMHYjqDayWQl4m_uuai9tQjIP60OTk8,886
 zenml/data_validators/base_data_validator.py,sha256=reGUJ6NEFfd_wocrcYBSU5QHCSBOll-pP_suTP_bfv0,9839
 zenml/entrypoints/__init__.py,sha256=2CMemOrHIJauxAss6k7dKFtsCFqgYR-JbAx4REoCaE8,946
-zenml/entrypoints/base_entrypoint_configuration.py,sha256=7l_AAj3qBPSj1fdjoc712RlhXa262BHyY4IlkDYiwA0,10332
+zenml/entrypoints/base_entrypoint_configuration.py,sha256=-JCuetzQGOPo5R6VmM2q2JVcLaEimrGPikTkXFXuiOI,10468
 zenml/entrypoints/entrypoint.py,sha256=XNgXBCMKoidmP0_AYgMpqo-neG8Y8jG0rj43ofTDZ9E,2033
 zenml/entrypoints/pipeline_entrypoint_configuration.py,sha256=To-vTP29qAE36ndJDF1fRw9wL2Nk2bsBuO-ayAwvSmo,1646
 zenml/entrypoints/step_entrypoint_configuration.py,sha256=fETr5E7eH8QEWYqgaarIrLQzrPrdLEr194xhd3H5yU4,7206
@@ -276,14 +276,14 @@ zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_oj
 zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=qPQOSacCFUcXCl0MkuzVkxpewU8NAPjiGQsjBF1N0ro,13492
 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=1SfFLoqF6ye005XhXjhtci-7fZ1Qd3_vzPIf2pmM49s,6658
+zenml/integrations/github/code_repositories/github_code_repository.py,sha256=7s4NjOn_bItAbwcxQQQfo9IlW8vowleUQMiBjjn6iJo,7041
 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=x5OSiWOsQprrleo8s8oIjgFZ7hz_tw66MLJYlSqONvc,17239
 zenml/integrations/github/plugins/github_webhook_event_source_flavor.py,sha256=jth8sxrmyg22-wT5Ax0fdsiLhTQwHXxaiTnB3kD97pk,1669
 zenml/integrations/gitlab/__init__.py,sha256=4Vz6XiPJYDZ9mos6L1FlgWsmueRCck86Sd8KRVj9NWQ,1003
 zenml/integrations/gitlab/code_repositories/__init__.py,sha256=Ds7NL6tCqLApRsOgvUofEq3Ms2No5_Z095uvi1gLVIk,817
-zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py,sha256=0XJhRBr6Ru31p5NDVbdTG57nXIAgWM630pLAWmJMVrw,5538
+zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py,sha256=P4boNw8ALFWWXrYiI9xG1cXiWhJXmoRoYmT8PQGBBBQ,5761
 zenml/integrations/great_expectations/__init__.py,sha256=Vp7qJ2iA3BfPK1QZFYEiE8SB8lIM2S8SGmW35dW8A7M,2448
 zenml/integrations/great_expectations/data_validators/__init__.py,sha256=Z16qmLfUoataEABQ6Ec-HSLM_a9VRALHFa4OoAyozIk,857
 zenml/integrations/great_expectations/data_validators/ge_data_validator.py,sha256=qp2ZFqQiYPszRc6vGhZhK22GEHhGoTQ0Y9u0trXNQyg,21404
@@ -648,7 +648,7 @@ zenml/models/v2/core/pipeline_deployment.py,sha256=6PbpnEUAbTT-_jPmyNtKuTJHFGdGL
 zenml/models/v2/core/pipeline_run.py,sha256=41TqY_CzUCWlunnyX6GBJ-OmcRtGIm9sHAXB4pIuwMo,32669
 zenml/models/v2/core/run_metadata.py,sha256=YDiPVRzTUMVHqW4T_Ke4GJMdscp0aS_g8I4P8n70szc,2436
 zenml/models/v2/core/run_template.py,sha256=A_dezNA9o1uJW-o1B9fYaSSWHwwglJYP3jvhWhGcJK4,12971
-zenml/models/v2/core/schedule.py,sha256=jcZSD-lbKxXxFL1lglQG5nFvY3-bLNarSrnmaxR42jY,10314
+zenml/models/v2/core/schedule.py,sha256=JNYTGPXDPMz7A9OjhnHIXyOLQuoFnrzOwChGibz2xQY,10774
 zenml/models/v2/core/secret.py,sha256=LoY5BMZIrU6KYa39layHt6cbAnmMR7yXizv_1Qqse6Q,11460
 zenml/models/v2/core/server_settings.py,sha256=al6LAXhoQHsBiTv2qlwFn1AlI2F8miogGB4xjDE6KeA,6267
 zenml/models/v2/core/service.py,sha256=lwCDOFk-OXmfhB_bOOJy7rhbkR40tCScQsWnllgL7II,14973
@@ -692,10 +692,10 @@ zenml/orchestrators/topsort.py,sha256=D8evz3X47zwpXd90NMLsJD-_uCeXtV6ClzNfDUrq7c
 zenml/orchestrators/utils.py,sha256=faRm86Ed_KVFBYbiMriSM0z3NwsydJWDvLYX-7DSrkc,13153
 zenml/orchestrators/wheeled_orchestrator.py,sha256=eOnMcnd3sCzfhA2l6qRAzF0rOXzaojbjvvYvTkqixQo,4791
 zenml/pipelines/__init__.py,sha256=hpIX7hN8jsQRHT5R-xSXZL88qrHwkmrvGLQeu1rWt4o,873
-zenml/pipelines/build_utils.py,sha256=Jas5D8QnjXzizMtcCrJY4dcddeix7QFXbhoY25R47M4,26133
+zenml/pipelines/build_utils.py,sha256=5eU1__3A37nRzYsNtExvtRfBHYXYOlDtcLS_cVT6zco,27255
 zenml/pipelines/pipeline_context.py,sha256=V_p-F9W7cBIlTjS0iv5-uJYMzaOj8bAUkc_uNhQgBms,3579
 zenml/pipelines/pipeline_decorator.py,sha256=FIbflYOMavbuyGmqsx3F5zZgg0oXMTi1eAcGXciljOs,4293
-zenml/pipelines/pipeline_definition.py,sha256=ByHnPiQWUdyy6UTMcKPXsbClTyFFO_geOGfOMTsRnt8,55924
+zenml/pipelines/pipeline_definition.py,sha256=bEloeIhfKjaKLG9zXccQ7Pb-KMRgNbVjVXeXC3X50HM,56034
 zenml/pipelines/run_utils.py,sha256=4KuHIQFtLXTZNQBScTEkIG5pqNtu6xGm6UZT7ptyyKs,11623
 zenml/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/plugins/base_plugin_flavor.py,sha256=88IxFW91UB_rQ8xPlfRnIhIJh7A308NEq2epMMdlOng,2530
@@ -756,12 +756,12 @@ zenml/types.py,sha256=LhGWJ4t3nybBk1l9Ox3tqqHbTYSuCqhkRsL5FqO6yf4,1206
 zenml/utils/__init__.py,sha256=jaMTbjm8tLYkaRoxlZ0Em4ye_ZHOHKgP2goPTTiYGUQ,797
 zenml/utils/archivable.py,sha256=QuLe1IhyABTrE6Y0hAT3tKjaUCpcze5ffZ_RKoKtJwY,6549
 zenml/utils/callback_registry.py,sha256=QBWdaraLAxBxi8DKbv9X1SUpTKDhhj-XE0JomB2Ax2Y,2411
-zenml/utils/code_repository_utils.py,sha256=CobRYMYfP2yNoA0hcu_WRz5oAff_jY95oyLCHz4fDOo,4734
+zenml/utils/code_repository_utils.py,sha256=f_VaN-QaCd66xVJJjM4mIo6Heu96-c9qdWh5XUjGpMY,4950
 zenml/utils/code_utils.py,sha256=y7_vmqYv_e11xekFjK7Dm4LkFu6SGl73tr2fVxBAK3s,11336
 zenml/utils/cuda_utils.py,sha256=RR21m__Zs-OnI5n-veFUzWniZjwLSbalHE5QV3jK1Hw,1624
 zenml/utils/daemon.py,sha256=GZ7Dx6GLHK04SR50wBxpKYmFhxPBfdLWxJiAWzJC6cM,11863
 zenml/utils/dashboard_utils.py,sha256=JyT-rOsqipazmImXBB-UzCg0GJ3UNzVCFdMcQisInfQ,5379
-zenml/utils/deprecation_utils.py,sha256=nianydl7n6DQrs_o5M0VxjYQfySvbGMPkjjdxn6aCvM,6253
+zenml/utils/deprecation_utils.py,sha256=QcWkOnzIRDKPOfkr523n3l2MoY2wE0LIPfbx99t4Gmg,6343
 zenml/utils/dict_utils.py,sha256=i7KAaKrkaWA_cG5IvVfMnr0CwWlBJ7KAsGvP2wxjZI8,2667
 zenml/utils/docker_utils.py,sha256=QvkKnvIYSKAhW7mErXwSaQ432-q1LAsLjo2YWSXD8Bk,13889
 zenml/utils/downloaded_repository_context.py,sha256=S660PSSQ3dsNBA0qAj8ap_Thyw1n6x4VDcRWbCMDP2M,1502
@@ -779,7 +779,7 @@ zenml/utils/networking_utils.py,sha256=zTDbOMkMPRWiWLZ2ccchd37rvvZWIIh0Kr9dZE-VJ
 zenml/utils/notebook_utils.py,sha256=VBMU9Qnx_NdpB1TQtnej56L0hRr11fwniOva3ltUT90,4113
 zenml/utils/package_utils.py,sha256=wy0Mh8hHhOX2z1JfGN5lifG9yEsBQGLwNfur0M3J2tQ,2730
 zenml/utils/pagination_utils.py,sha256=TufckOqOKeDPwE3ySefL05zOzGUUA2Fqx_QFVhE2so0,1445
-zenml/utils/pipeline_docker_image_builder.py,sha256=NzHWv0hi2U5d3OsOlTea1fdeA0C4hNMJ3IqUqDsB4r4,25172
+zenml/utils/pipeline_docker_image_builder.py,sha256=5gTY5bf58UYy1p_s7_4odEcunC7Hpe1DbdVnCFyPkiY,25034
 zenml/utils/proxy_utils.py,sha256=fgRlLa9pfXJDoxtB31_YP7DClOMQLek_nMmM0et6i3w,7241
 zenml/utils/pydantic_utils.py,sha256=oQcxY4VXuVY3n632atlvdmi12EYcSQ1xZuQJY3Je-sA,16592
 zenml/utils/requirements_utils.py,sha256=pUVlQpEtLfz7lLJEUN-t7oHKLzdZZdgHoMzv0V5WXZI,2250
@@ -1253,7 +1253,7 @@ zenml/zen_stores/schemas/api_key_schemas.py,sha256=d_ewTrIJpRjLWupOZFpWcPSlRiHxi
 zenml/zen_stores/schemas/artifact_schemas.py,sha256=_hXytHqa0BYrBLgZt8RdAid1n291d8wQ0HxNMIAjlJQ,14317
 zenml/zen_stores/schemas/artifact_visualization_schemas.py,sha256=_gMNjOF0oJ0ruW5Ua86hDocvO4MT3ACAIx_BwFBJFWk,3693
 zenml/zen_stores/schemas/base_schemas.py,sha256=-MDYnS4ZLDe6x75nTXuVrOSbcYOuECLNk-yxjfEdJh4,2043
-zenml/zen_stores/schemas/code_repository_schemas.py,sha256=aDot3lUrcjPVX7yDlCzMuMG9ziX6vDSZwfHKtNbNlIc,7410
+zenml/zen_stores/schemas/code_repository_schemas.py,sha256=K2EFvu_bkF9-cDAVYpMntNidSu-zb3By7CVq6rLj97c,7489
 zenml/zen_stores/schemas/component_schemas.py,sha256=vQ0GHbtvx-rgJDMPtIIbLs_KM8pRfNnF2EL41CYx48A,8260
 zenml/zen_stores/schemas/constants.py,sha256=bcadtiWEWkpzRzvnj46yuAaC7DE09g2H9L9r398lV00,704
 zenml/zen_stores/schemas/device_schemas.py,sha256=voVBknm14NmFpij1EGb_HhVlUng0cglCMXkYSxDRq1s,9126
@@ -1292,8 +1292,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Bq1djrUP9saoD7vECjS7
 zenml/zen_stores/sql_zen_store.py,sha256=AaPwO6R_2QVsI_PmDktVTF-Fx6VzR64j-1oU7cJwVxI,416486
 zenml/zen_stores/template_utils.py,sha256=EKYBgmDLTS_PSMWaIO5yvHPLiQvMqHcsAe6NUCrv-i4,9068
 zenml/zen_stores/zen_store_interface.py,sha256=vf2gKBWfUUPtcGZC35oQB6pPNVzWVyQC8nWxVLjfrxM,92692
-zenml_nightly-0.73.0.dev20250127.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.73.0.dev20250127.dist-info/METADATA,sha256=hQqzjMYXjxVPekLNRFGO9TI6yCO0KMMPTRR4XZccL6o,21355
-zenml_nightly-0.73.0.dev20250127.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-zenml_nightly-0.73.0.dev20250127.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.73.0.dev20250127.dist-info/RECORD,,
+zenml_nightly-0.73.0.dev20250128.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.73.0.dev20250128.dist-info/METADATA,sha256=BV3mpMHnzE1NHGrXN1GGMGemkWe14SU-ysUGOXW2kg8,21355
+zenml_nightly-0.73.0.dev20250128.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+zenml_nightly-0.73.0.dev20250128.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.73.0.dev20250128.dist-info/RECORD,,