zenml-nightly 0.75.0.dev20250310__py3-none-any.whl → 0.75.0.dev20250312__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.75.0.dev20250310
+0.75.0.dev20250312

zenml/cli/utils.py

@@ -1419,7 +1419,7 @@ def describe_pydantic_object(schema_json: Dict[str, Any]) -> None:
                     prop_type = prop_schema["type"]
                 elif "anyOf" in prop_schema.keys():
                     prop_type = ", ".join(
-                        [p["type"] for p in prop_schema["anyOf"]]
+                        [p.get("type", "object") for p in prop_schema["anyOf"]]
                     )
                     prop_type = f"one of: {prop_type}"
                 else:

zenml/client.py

@@ -2674,6 +2674,7 @@ class Client(metaclass=ClientMetaClass):
         python_version: Optional[str] = None,
         checksum: Optional[str] = None,
         stack_checksum: Optional[str] = None,
+        duration: Optional[Union[int, str]] = None,
         hydrate: bool = False,
     ) -> Page[PipelineBuildResponse]:
         """List all builds.
@@ -2699,6 +2700,7 @@ class Client(metaclass=ClientMetaClass):
             python_version: The Python version to filter by.
             checksum: The build checksum to filter by.
             stack_checksum: The stack checksum to filter by.
+            duration: The duration of the build in seconds to filter by.
             hydrate: Flag deciding whether to hydrate the output model(s)
                 by including metadata fields in the response.
 
@@ -2725,6 +2727,7 @@ class Client(metaclass=ClientMetaClass):
             python_version=python_version,
             checksum=checksum,
             stack_checksum=stack_checksum,
+            duration=duration,
         )
         build_filter_model.set_scope_workspace(self.active_workspace.id)
         return self.zen_store.list_builds(

zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py

@@ -54,7 +54,7 @@ class GitLabCodeRepositoryConfig(BaseCodeRepositoryConfig):
     group: str
     project: str
     host: Optional[str] = "gitlab.com"
-    token: str = SecretField()
+    token: Optional[str] = SecretField(default=None)
 
     url: Optional[str] = None
     _deprecation_validator = deprecation_utils.deprecate_pydantic_attributes(
@@ -109,10 +109,11 @@ class GitLabCodeRepository(BaseCodeRepository):
             self._gitlab_session = Gitlab(
                 url=self.config.instance_url, private_token=self.config.token
             )
-            self._gitlab_session.auth()
-            user = self._gitlab_session.user or None
-            if user:
-                logger.debug(f"Logged in as {user.username}")
+
+            if self.config.token:
+                self._gitlab_session.auth()
+                if user := self._gitlab_session.user:
+                    logger.debug(f"Logged into GitLab as {user.username}")
         except Exception as e:
             raise RuntimeError(f"An error occurred while logging in: {str(e)}")
 

zenml/models/v2/core/pipeline_build.py

@@ -72,6 +72,9 @@ class PipelineBuildBase(BaseZenModel):
     python_version: Optional[str] = Field(
         title="The Python version used for this build.", default=None
     )
+    duration: Optional[int] = Field(
+        title="The duration of the build in seconds.", default=None
+    )
 
     # Helper methods
     @property
@@ -223,6 +226,9 @@ class PipelineBuildResponseMetadata(WorkspaceScopedResponseMetadata):
     contains_code: bool = Field(
         title="Whether any image of the build contains user code.",
     )
+    duration: Optional[int] = Field(
+        title="The duration of the build in seconds.", default=None
+    )
 
 
 class PipelineBuildResponseResources(WorkspaceScopedResponseResources):
@@ -454,6 +460,15 @@ class PipelineBuildResponse(
         """
         return self.get_metadata().contains_code
 
+    @property
+    def duration(self) -> Optional[int]:
+        """The `duration` property.
+
+        Returns:
+            the value of the property.
+        """
+        return self.get_metadata().duration
+
 
 # ------------------ Filter Model ------------------
 
@@ -502,6 +517,9 @@ class PipelineBuildFilter(WorkspaceScopedFilter):
     stack_checksum: Optional[str] = Field(
         description="The stack checksum.", default=None
     )
+    duration: Optional[Union[int, str]] = Field(
+        description="The duration of the build in seconds.", default=None
+    )
 
     def get_custom_filters(
         self,

zenml/pipelines/build_utils.py

@@ -15,6 +15,7 @@
 
 import hashlib
 import platform
+import time
 from typing import (
     TYPE_CHECKING,
     Dict,
@@ -327,6 +328,7 @@ def create_pipeline_build(
         "Building Docker image(s) for pipeline `%s`.",
         deployment.pipeline_configuration.name,
     )
+    start_time = time.time()
 
     docker_image_builder = PipelineDockerImageBuilder()
     images: Dict[str, BuildItem] = {}
@@ -409,6 +411,7 @@ def create_pipeline_build(
 
     logger.info("Finished building Docker image(s).")
 
+    duration = round(time.time() - start_time)
     is_local = stack.container_registry is None
     contains_code = any(item.contains_code for item in images.values())
     build_checksum = compute_build_checksum(
@@ -427,6 +430,7 @@ def create_pipeline_build(
         python_version=platform.python_version(),
         checksum=build_checksum,
         stack_checksum=stack_checksum,
+        duration=duration,
     )
     return client.zen_store.create_build(build_request)
 

zenml/zen_stores/migrations/versions/0392807467dc_add_build_duration.py

@@ -0,0 +1,34 @@
+"""Add build duration [0392807467dc].
+
+Revision ID: 0392807467dc
+Revises: 0.75.0
+Create Date: 2025-02-28 15:50:06.960529
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "0392807467dc"
+down_revision = "0.75.0"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("pipeline_build", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("duration", sa.Integer(), nullable=True))
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("pipeline_build", schema=None) as batch_op:
+        batch_op.drop_column("duration")
+
+    # ### end Alembic commands ###

zenml/zen_stores/schemas/pipeline_build_schemas.py

@@ -100,6 +100,8 @@ class PipelineBuildSchema(BaseSchema, table=True):
     python_version: Optional[str]
     checksum: Optional[str]
     stack_checksum: Optional[str]
+    # Build duration in seconds
+    duration: Optional[int] = None
 
     @classmethod
     def from_request(
@@ -125,6 +127,7 @@ class PipelineBuildSchema(BaseSchema, table=True):
             python_version=request.python_version,
             checksum=request.checksum,
             stack_checksum=request.stack_checksum,
+            duration=request.duration,
         )
 
     def to_model(
@@ -162,6 +165,7 @@ class PipelineBuildSchema(BaseSchema, table=True):
                 stack_checksum=self.stack_checksum,
                 is_local=self.is_local,
                 contains_code=self.contains_code,
+                duration=self.duration,
             )
         return PipelineBuildResponse(
             id=self.id,

{zenml_nightly-0.75.0.dev20250310.dist-info → zenml_nightly-0.75.0.dev20250312.dist-info}/METADATA

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

{zenml_nightly-0.75.0.dev20250310.dist-info → zenml_nightly-0.75.0.dev20250312.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=XnOrqbxLYtJOamCOfqZBBLGbgglmmBMCxMXAI2D8IAk,19
+zenml/VERSION,sha256=WKPWjLdCyQPwGFHDNY91sE5KrdVWKRkJNQkmmNhO_DI,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
@@ -52,10 +52,10 @@ zenml/cli/stack_components.py,sha256=7RXVCJTX6sU-CiyDeu4kGYPJZ7QRQjDTptsvl__sI_A
 zenml/cli/tag.py,sha256=Co-AL1eNZMf_AsqlcnJpoZ_k7UKa0l5X_g_TLpJdGes,4884
 zenml/cli/text_utils.py,sha256=bY1GIjoULt1cW2FyrPlMoAXNS2R7cSOjDFEZQqrpVQ8,3553
 zenml/cli/user_management.py,sha256=fTuRworQahst_j78qPYTtgciUeUOxwo7efiyPwmj2tI,13075
-zenml/cli/utils.py,sha256=oQm8clYsuATzEjXJNVMwmdTiygOwxErvUGONPIxb7no,86478
+zenml/cli/utils.py,sha256=ktTScIfvbm_RTSbw6ydlZT2CbjRm_sUBzON4wqF1WqY,86492
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
 zenml/cli/workspace.py,sha256=bp02aXou574ToWPD8OAIB_cg3mvpE011H8aMKegT-nU,2970
-zenml/client.py,sha256=VGpvUNfs6D_7JthAjocEO6pDNfjeNkWwNudWVYrbgqo,283571
+zenml/client.py,sha256=24HZcdRn44-R_RBkBh-nQ-DdzvibcQkvKMyekQFiOyc,283727
 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=Id6VjbUu8N3ZpNvBGhOgbahtoMiCAtYXed3G7YQ_iAc,5225
@@ -283,7 +283,7 @@ zenml/integrations/github/plugins/event_sources/github_webhook_event_source.py,s
 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=1i4XLkHoaR1ap5ZlQOkfG2G_-mkXLs3yQmvPABgdK5k,6421
+zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py,sha256=41Alp4mzdofEI_SejQXfrDTQ8L2w_FiU1rYOgyxEUhg,6475
 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
@@ -643,7 +643,7 @@ zenml/models/v2/core/model_version.py,sha256=fkERxxFR_08LosUdbPF5hIeoz0n4C0703f5
 zenml/models/v2/core/model_version_artifact.py,sha256=ZyupuEsfPSjgaHDIUOojwJepJt1LTXDoUyE2h9htB_s,9086
 zenml/models/v2/core/model_version_pipeline_run.py,sha256=JbPZZEQvOK9I32htkWdAONy7gvtoz_Jo45B5gYrh2vs,7036
 zenml/models/v2/core/pipeline.py,sha256=cql05iz2jqrlyRfn3jB4iQ_AgpHapD6Sr5Kd_6Q1W30,12044
-zenml/models/v2/core/pipeline_build.py,sha256=bjSvmNnTF9xAycYfmTHO2wzgR-ZVWyy5V5-9ajZUxag,16399
+zenml/models/v2/core/pipeline_build.py,sha256=X3aK0HzmyUueAqSCw9sykLPOOWJKua8U8JacOuTY7NQ,16955
 zenml/models/v2/core/pipeline_deployment.py,sha256=6PbpnEUAbTT-_jPmyNtKuTJHFGdGLD2h5Zu_pXIHW1I,12108
 zenml/models/v2/core/pipeline_run.py,sha256=mmr3EP5jWq7sQ5FnUTD4kXYWY-yF5x6SW9Ei2a78Stc,33195
 zenml/models/v2/core/run_metadata.py,sha256=YDiPVRzTUMVHqW4T_Ke4GJMdscp0aS_g8I4P8n70szc,2436
@@ -692,7 +692,7 @@ 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=DkID1YnRYkw569uU-gidpF8WR8E7K55_wn3CzRPQ3Cs,27562
+zenml/pipelines/build_utils.py,sha256=S5lTxzScaYGiJuC8ytf26f3FIB46y5kqN5OzxmfxnNI,27677
 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=pl5tV4_h03MQDaIJZQCJJ8mY5S3Js-89r5NFIBpxB6A,57244
@@ -1150,6 +1150,7 @@ zenml/zen_stores/migrations/versions/0.74.0_release.py,sha256=G21sCXOaWXDLLMQHs7
 zenml/zen_stores/migrations/versions/0.75.0_release.py,sha256=dRlxBHwDGALjM5whROu145SQ6tDocjJ29XBWQC4bxFw,450
 zenml/zen_stores/migrations/versions/026d4577b6a0_add_code_path.py,sha256=hXLzvQcylNrbCVD6vha52PFkSPNC2klW9kA0vuQX_cE,1091
 zenml/zen_stores/migrations/versions/03742aa7fdd7_add_secrets.py,sha256=gewKqu1AnzvNTjVvK1eaAwP0hVneWDUyDRSLTvRCdpg,1587
+zenml/zen_stores/migrations/versions/0392807467dc_add_build_duration.py,sha256=YlkDBlfBBv45FsrMO11YcdRn4Maqmlg77t8gWJO4DfA,982
 zenml/zen_stores/migrations/versions/0701da9951a0_added_service_table.py,sha256=dmMUbQhcQparYdV9nfxcgcJTwlEbUWJlzMQ9d0YQyrs,3399
 zenml/zen_stores/migrations/versions/0b06faa59c93_add_service_connectors.py,sha256=z7Bz5jVP0MVSX3__GN0CtZZhI_t88gcgZzNpPxsIJl4,4400
 zenml/zen_stores/migrations/versions/0d707865f404_adding_labels_to_stacks.py,sha256=l9WImB9dM5X2CfU4Fd_tzTNo2-cP2xTFg9FongasMYc,787
@@ -1258,7 +1259,7 @@ zenml/zen_stores/schemas/event_source_schemas.py,sha256=qhkyKg3zl7pFaxDYcXsDvJEI
 zenml/zen_stores/schemas/flavor_schemas.py,sha256=cEKHrQp4fr16SDjOqBubM-jAbq4dK3MVyZSe-70e-ME,5054
 zenml/zen_stores/schemas/logs_schemas.py,sha256=qv6fs3JiVgzlmTXJqb_gG5NsU5q_50e0167JiWIxR14,3588
 zenml/zen_stores/schemas/model_schemas.py,sha256=95q5OaQ9DFmTno76wLaLmeBj3cePrVju20flWiJHPOk,25460
-zenml/zen_stores/schemas/pipeline_build_schemas.py,sha256=QK3sVPYhumUx_x-H3YZ-RfnsxGhKfkMGdP5FC9WrQgU,5718
+zenml/zen_stores/schemas/pipeline_build_schemas.py,sha256=IN5vHCzZ6ULUN1bG15O5-B-lQLpjTYwnT1MIUnlDfww,5864
 zenml/zen_stores/schemas/pipeline_deployment_schemas.py,sha256=fdsROPBE1QrWKvSkaZYz2Sc1MwXogAembs8TfZML3ks,10201
 zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=ylodIinwKGqKCIcZaKJwiA27g0rkNjKY3DLx_E2lylo,18046
 zenml/zen_stores/schemas/pipeline_schemas.py,sha256=tUgBics0QOszIN870_-ADcBHINH8v32-CLjPW0vHzMg,7159
@@ -1289,8 +1290,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=L3PtBT-VULJVh4udD4eY4TbU_o1iEC23VyUh26uBo5Y,415902
 zenml/zen_stores/template_utils.py,sha256=EKYBgmDLTS_PSMWaIO5yvHPLiQvMqHcsAe6NUCrv-i4,9068
 zenml/zen_stores/zen_store_interface.py,sha256=vf2gKBWfUUPtcGZC35oQB6pPNVzWVyQC8nWxVLjfrxM,92692
-zenml_nightly-0.75.0.dev20250310.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.75.0.dev20250310.dist-info/METADATA,sha256=iePs9jYfQIsL6LLZSbkaphHjKJ3HJGKNES5PV6XkyuE,24274
-zenml_nightly-0.75.0.dev20250310.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-zenml_nightly-0.75.0.dev20250310.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.75.0.dev20250310.dist-info/RECORD,,
+zenml_nightly-0.75.0.dev20250312.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.75.0.dev20250312.dist-info/METADATA,sha256=mmZp_NSq_xJgmAPLEJIkZHfoGiUNb4wU8--2380rMqA,24274
+zenml_nightly-0.75.0.dev20250312.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+zenml_nightly-0.75.0.dev20250312.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.75.0.dev20250312.dist-info/RECORD,,