zenml-nightly 0.82.1.dev20250520__tar.gz → 0.82.1.dev20250522__tar.gz

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/PKG-INFO

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

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zenml-nightly"
-version = "0.82.1.dev20250520"
+version = "0.82.1.dev20250522"
 packages = [{ include = "zenml", from = "src" }]
 description = "ZenML: Write production-ready ML code."
 authors = ["ZenML GmbH <info@zenml.io>"]

zenml_nightly-0.82.1.dev20250522/src/zenml/VERSION

@@ -0,0 +1 @@
+0.82.1.dev20250522

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/config/build_configuration.py

@@ -108,6 +108,13 @@ class BuildConfiguration(BaseModel):
             )
             if digest:
                 hash_.update(digest.encode())
+            elif self.settings.skip_build:
+                # If the user has specified to skip the build, the image being
+                # used for the run is whatever they set for `parent_image`.
+                # In this case, the bevavior depends on the orchestrators image
+                # pull policy, and whether there is any caching involved. This
+                # is out of our control here, so we don't log any warning.
+                pass
             else:
                 logger.warning(
                     "Unable to fetch parent image digest for image `%s`. "

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/config/schedule.py

@@ -108,7 +108,7 @@ class Schedule(BaseModel):
         if self.cron_expression and periodic_schedule:
             logger.warning(
                 "This schedule was created with a cron expression as well as "
-                "values for `start_time` and `interval_seconds`. The resulting "
+                "values for `start_time` and `interval_second`. The resulting "
                 "behavior depends on the concrete orchestrator implementation "
                 "but will usually ignore the interval and use the cron "
                 "expression."

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/services/local/local_service.py

@@ -414,7 +414,8 @@ class LocalDaemonService(BaseService):
             p = psutil.Process(pid)
         except psutil.Error:
             logger.error(
-                "Could not find process for for service '%s' ...", self
+                "Could not find process for service '%s' ...",
+                self,
             )
             return
         if force:

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/utils/yaml_utils.py

@@ -14,6 +14,7 @@
 """Utility functions to help with YAML files and data."""
 
 import json
+import os
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Type, Union
 from uuid import UUID
@@ -99,9 +100,8 @@ def is_yaml(file_path: str) -> bool:
     Returns:
         True if is yaml, else False.
     """
-    if file_path.endswith("yaml") or file_path.endswith("yml"):
-        return True
-    return False
+    extension = os.path.splitext(file_path)[1].lower()
+    return extension in (".yaml", ".yml")
 
 
 def comment_out_yaml(yaml_string: str) -> str:

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/auth.py

@@ -434,17 +434,10 @@ def authenticate_credentials(
                     run does not exist.
                 """
                 try:
-                    pipeline_run = zen_store().get_run(
-                        pipeline_run_id, hydrate=True
-                    )
+                    return zen_store().get_run_status(pipeline_run_id)
                 except KeyError:
                     return None, None
 
-                return (
-                    pipeline_run.status,
-                    pipeline_run.end_time,
-                )
-
             (
                 pipeline_run_status,
                 pipeline_run_end_time,

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/actions_endpoints.py

@@ -44,7 +44,7 @@ from zenml.zen_server.rbac.utils import (
     verify_permission_for_model,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     plugin_flavor_registry,
     zen_store,
@@ -61,7 +61,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_actions(
     action_filter_model: ActionFilter = Depends(make_dependable(ActionFilter)),
     hydrate: bool = False,
@@ -132,7 +132,7 @@ def list_actions(
     "/{action_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_action(
     action_id: UUID,
     hydrate: bool = True,
@@ -179,7 +179,7 @@ def get_action(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_action(
     action: ActionRequest,
     _: AuthContext = Security(authorize),
@@ -225,7 +225,7 @@ def create_action(
     "/{action_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_action(
     action_id: UUID,
     action_update: ActionUpdate,
@@ -280,7 +280,7 @@ def update_action(
     "/{action_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_action(
     action_id: UUID,
     force: bool = False,

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/artifact_endpoint.py

@@ -36,7 +36,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 )
 from zenml.zen_server.rbac.models import ResourceType
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -52,7 +52,7 @@ artifact_router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_artifacts(
     artifact_filter_model: ArtifactFilter = Depends(
         make_dependable(ArtifactFilter)
@@ -83,7 +83,7 @@ def list_artifacts(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_artifact(
     artifact: ArtifactRequest,
     _: AuthContext = Security(authorize),
@@ -106,7 +106,7 @@ def create_artifact(
     "/{artifact_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact(
     artifact_id: UUID,
     hydrate: bool = True,
@@ -133,7 +133,7 @@ def get_artifact(
     "/{artifact_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_artifact(
     artifact_id: UUID,
     artifact_update: ArtifactUpdate,
@@ -160,7 +160,7 @@ def update_artifact(
     "/{artifact_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_artifact(
     artifact_id: UUID,
     _: AuthContext = Security(authorize),

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/artifact_version_endpoints.py

@@ -66,7 +66,7 @@ from zenml.zen_server.rbac.utils import (
     get_allowed_resource_ids,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     set_filter_project_scope,
     zen_store,
@@ -83,7 +83,7 @@ artifact_version_router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_artifact_versions(
     artifact_version_filter_model: ArtifactVersionFilter = Depends(
         make_dependable(ArtifactVersionFilter)
@@ -127,7 +127,7 @@ def list_artifact_versions(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_artifact_version(
     artifact_version: ArtifactVersionRequest,
     _: AuthContext = Security(authorize),
@@ -150,7 +150,7 @@ def create_artifact_version(
     BATCH,
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def batch_create_artifact_version(
     artifact_versions: List[ArtifactVersionRequest],
     _: AuthContext = Security(authorize),
@@ -173,7 +173,7 @@ def batch_create_artifact_version(
     "/{artifact_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact_version(
     artifact_version_id: UUID,
     hydrate: bool = True,
@@ -200,7 +200,7 @@ def get_artifact_version(
     "/{artifact_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_artifact_version(
     artifact_version_id: UUID,
     artifact_version_update: ArtifactVersionUpdate,
@@ -227,7 +227,7 @@ def update_artifact_version(
     "/{artifact_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_artifact_version(
     artifact_version_id: UUID,
     _: AuthContext = Security(authorize),
@@ -248,7 +248,7 @@ def delete_artifact_version(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def prune_artifact_versions(
     project_name_or_id: Union[str, UUID],
     only_versions: bool = True,
@@ -275,7 +275,7 @@ def prune_artifact_versions(
     "/{artifact_version_id}" + VISUALIZE,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact_visualization(
     artifact_version_id: UUID,
     index: int = 0,
@@ -303,7 +303,7 @@ def get_artifact_visualization(
     "/{artifact_version_id}" + DOWNLOAD_TOKEN,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact_download_token(
     artifact_version_id: UUID,
     _: AuthContext = Security(authorize),
@@ -334,7 +334,7 @@ def get_artifact_download_token(
     "/{artifact_version_id}" + DATA,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def download_artifact_data(
     artifact_version_id: UUID, token: str
 ) -> FileResponse:

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/auth_endpoints.py

@@ -71,8 +71,8 @@ from zenml.zen_server.rbac.utils import (
     verify_permission_for_model,
 )
 from zenml.zen_server.utils import (
+    async_fastapi_endpoint_wrapper,
     get_ip_location,
-    handle_exceptions,
     server_config,
     zen_store,
 )
@@ -222,11 +222,11 @@ class OAuthLoginRequestForm:
     LOGIN,
     response_model=Union[OAuthTokenResponse, OAuthRedirectResponse],
 )
+@async_fastapi_endpoint_wrapper
 @rate_limit_requests(
     day_limit=server_config().login_rate_limit_day,
     minute_limit=server_config().login_rate_limit_minute,
 )
-@handle_exceptions
 def token(
     request: Request,
     response: Response,
@@ -341,6 +341,7 @@ def logout(
     DEVICE_AUTHORIZATION,
     response_model=OAuthDeviceAuthorizationResponse,
 )
+@async_fastapi_endpoint_wrapper
 def device_authorization(
     request: Request,
     client_id: UUID = Form(...),
@@ -471,7 +472,7 @@ def device_authorization(
     API_TOKEN,
     response_model=str,
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def api_token(
     token_type: APITokenType = APITokenType.GENERIC,
     expires_in: Optional[int] = None,

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/code_repositories_endpoints.py

@@ -38,7 +38,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 from zenml.zen_server.rbac.models import ResourceType
 from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -62,7 +62,7 @@ router = APIRouter(
     deprecated=True,
     tags=["code_repositories"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_code_repository(
     code_repository: CodeRepositoryRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
@@ -99,7 +99,7 @@ def create_code_repository(
     deprecated=True,
     tags=["code_repositories"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_code_repositories(
     filter_model: CodeRepositoryFilter = Depends(
         make_dependable(CodeRepositoryFilter)
@@ -135,7 +135,7 @@ def list_code_repositories(
     "/{code_repository_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_code_repository(
     code_repository_id: UUID,
     hydrate: bool = True,
@@ -162,7 +162,7 @@ def get_code_repository(
     "/{code_repository_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_code_repository(
     code_repository_id: UUID,
     update: CodeRepositoryUpdate,
@@ -189,7 +189,7 @@ def update_code_repository(
     "/{code_repository_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_code_repository(
     code_repository_id: UUID,
     _: AuthContext = Security(authorize),

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/devices_endpoints.py

@@ -39,7 +39,7 @@ from zenml.utils.time_utils import utc_now
 from zenml.zen_server.auth import AuthContext, authorize
 from zenml.zen_server.exceptions import error_response
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     server_config,
     zen_store,
@@ -56,7 +56,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_authorized_devices(
     filter_model: OAuthDeviceFilter = Depends(
         make_dependable(OAuthDeviceFilter)
@@ -86,7 +86,7 @@ def list_authorized_devices(
     "/{device_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_authorization_device(
     device_id: UUID,
     user_code: Optional[str] = None,
@@ -136,7 +136,7 @@ def get_authorization_device(
     "/{device_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_authorized_device(
     device_id: UUID,
     update: OAuthDeviceUpdate,
@@ -172,7 +172,7 @@ def update_authorized_device(
     "/{device_id}" + DEVICE_VERIFY,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def verify_authorized_device(
     device_id: UUID,
     request: OAuthDeviceVerificationRequest,
@@ -276,7 +276,7 @@ def verify_authorized_device(
     "/{device_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_authorized_device(
     device_id: UUID,
     auth_context: AuthContext = Security(authorize),

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/event_source_endpoints.py

@@ -39,7 +39,7 @@ from zenml.zen_server.rbac.utils import (
     verify_permission_for_model,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     plugin_flavor_registry,
     zen_store,
@@ -59,7 +59,7 @@ event_source_router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_event_sources(
     event_source_filter_model: EventSourceFilter = Depends(
         make_dependable(EventSourceFilter)
@@ -133,7 +133,7 @@ def list_event_sources(
     "/{event_source_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_event_source(
     event_source_id: UUID,
     hydrate: bool = True,
@@ -185,7 +185,7 @@ def get_event_source(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_event_source(
     event_source: EventSourceRequest,
     _: AuthContext = Security(authorize),
@@ -227,7 +227,7 @@ def create_event_source(
     "/{event_source_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_event_source(
     event_source_id: UUID,
     event_source_update: EventSourceUpdate,
@@ -279,7 +279,7 @@ def update_event_source(
     "/{event_source_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_event_source(
     event_source_id: UUID,
     force: bool = False,

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/flavors_endpoints.py

@@ -37,7 +37,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 from zenml.zen_server.rbac.models import Action, ResourceType
 from zenml.zen_server.rbac.utils import verify_permission
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -53,7 +53,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_flavors(
     flavor_filter_model: FlavorFilter = Depends(make_dependable(FlavorFilter)),
     hydrate: bool = False,
@@ -82,7 +82,7 @@ def list_flavors(
     "/{flavor_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_flavor(
     flavor_id: UUID,
     hydrate: bool = True,
@@ -107,7 +107,7 @@ def get_flavor(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_flavor(
     flavor: FlavorRequest,
     _: AuthContext = Security(authorize),
@@ -130,7 +130,7 @@ def create_flavor(
     "/{flavor_id}",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_flavor(
     flavor_id: UUID,
     flavor_update: FlavorUpdate,
@@ -159,7 +159,7 @@ def update_flavor(
     "/{flavor_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_flavor(
     flavor_id: UUID,
     _: AuthContext = Security(authorize),
@@ -180,7 +180,7 @@ def delete_flavor(
     "/sync",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def sync_flavors(
     _: AuthContext = Security(authorize),
 ) -> None:

{zenml_nightly-0.82.1.dev20250520 → zenml_nightly-0.82.1.dev20250522}/src/zenml/zen_server/routers/logs_endpoints.py

@@ -29,7 +29,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_get_entity,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     zen_store,
 )
 
@@ -44,7 +44,7 @@ router = APIRouter(
     "/{logs_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_logs(
     logs_id: UUID,
     hydrate: bool = True,