zenml-nightly 0.75.0.dev20250313__py3-none-any.whl → 0.75.0.dev20250314__py3-none-any.whl

zenml/models/v2/core/model.py

@@ -25,13 +25,13 @@ from zenml.constants import (
 )
 from zenml.models.v2.base.base import BaseUpdate
 from zenml.models.v2.base.scoped import (
+    ProjectScopedFilter,
+    ProjectScopedRequest,
+    ProjectScopedResponse,
+    ProjectScopedResponseBody,
+    ProjectScopedResponseMetadata,
+    ProjectScopedResponseResources,
     TaggableFilter,
-    WorkspaceScopedFilter,
-    WorkspaceScopedRequest,
-    WorkspaceScopedResponse,
-    WorkspaceScopedResponseBody,
-    WorkspaceScopedResponseMetadata,
-    WorkspaceScopedResponseResources,
 )
 from zenml.utils.pagination_utils import depaginate
 
@@ -47,7 +47,7 @@ AnyQuery = TypeVar("AnyQuery", bound=Any)
 # ------------------ Request Model ------------------
 
 
-class ModelRequest(WorkspaceScopedRequest):
+class ModelRequest(ProjectScopedRequest):
     """Request model for models."""
 
     name: str = Field(
@@ -121,7 +121,7 @@ class ModelUpdate(BaseUpdate):
 # ------------------ Response Model ------------------
 
 
-class ModelResponseBody(WorkspaceScopedResponseBody):
+class ModelResponseBody(ProjectScopedResponseBody):
     """Response body for models."""
 
     tags: List["TagResponse"] = Field(
@@ -131,7 +131,7 @@ class ModelResponseBody(WorkspaceScopedResponseBody):
     latest_version_id: Optional[UUID] = None
 
 
-class ModelResponseMetadata(WorkspaceScopedResponseMetadata):
+class ModelResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for models."""
 
     license: Optional[str] = Field(
@@ -175,12 +175,12 @@ class ModelResponseMetadata(WorkspaceScopedResponseMetadata):
     )
 
 
-class ModelResponseResources(WorkspaceScopedResponseResources):
+class ModelResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the model entity."""
 
 
 class ModelResponse(
-    WorkspaceScopedResponse[
+    ProjectScopedResponse[
         ModelResponseBody, ModelResponseMetadata, ModelResponseResources
     ]
 ):
@@ -315,7 +315,7 @@ class ModelResponse(
         model_versions = depaginate(
             client.list_model_versions,
             model_name_or_id=self.id,
-            workspace=self.workspace.id,
+            project=self.project.id,
         )
         return [
             mv.to_model_class(suppress_class_validation_warnings=True)
@@ -326,8 +326,8 @@ class ModelResponse(
 # ------------------ Filter Model ------------------
 
 
-class ModelFilter(WorkspaceScopedFilter, TaggableFilter):
-    """Model to enable advanced filtering of all Workspaces."""
+class ModelFilter(ProjectScopedFilter, TaggableFilter):
+    """Model to enable advanced filtering of all models."""
 
     name: Optional[str] = Field(
         default=None,
@@ -335,16 +335,16 @@ class ModelFilter(WorkspaceScopedFilter, TaggableFilter):
     )
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.FILTER_EXCLUDE_FIELDS,
         *TaggableFilter.FILTER_EXCLUDE_FIELDS,
     ]
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *ProjectScopedFilter.CUSTOM_SORTING_OPTIONS,
         *TaggableFilter.CUSTOM_SORTING_OPTIONS,
         SORT_BY_LATEST_VERSION_KEY,
     ]
     CLI_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.CLI_EXCLUDE_FIELDS,
         *TaggableFilter.CLI_EXCLUDE_FIELDS,
     ]
 

zenml/models/v2/core/model_version.py

@@ -34,13 +34,13 @@ from zenml.models.v2.base.base import BaseUpdate
 from zenml.models.v2.base.filter import AnyQuery
 from zenml.models.v2.base.page import Page
 from zenml.models.v2.base.scoped import (
+    ProjectScopedFilter,
+    ProjectScopedRequest,
+    ProjectScopedResponse,
+    ProjectScopedResponseBody,
+    ProjectScopedResponseMetadata,
+    ProjectScopedResponseResources,
     TaggableFilter,
-    WorkspaceScopedFilter,
-    WorkspaceScopedRequest,
-    WorkspaceScopedResponse,
-    WorkspaceScopedResponseBody,
-    WorkspaceScopedResponseMetadata,
-    WorkspaceScopedResponseResources,
 )
 from zenml.models.v2.core.service import ServiceResponse
 from zenml.models.v2.core.tag import TagResponse
@@ -60,7 +60,7 @@ if TYPE_CHECKING:
 # ------------------ Request Model ------------------
 
 
-class ModelVersionRequest(WorkspaceScopedRequest):
+class ModelVersionRequest(ProjectScopedRequest):
     """Request model for model versions."""
 
     name: Optional[str] = Field(
@@ -135,7 +135,7 @@ class ModelVersionUpdate(BaseUpdate):
 # ------------------ Response Model ------------------
 
 
-class ModelVersionResponseBody(WorkspaceScopedResponseBody):
+class ModelVersionResponseBody(ProjectScopedResponseBody):
     """Response body for model versions."""
 
     stage: Optional[str] = Field(
@@ -178,7 +178,7 @@ class ModelVersionResponseBody(WorkspaceScopedResponseBody):
     model_config = ConfigDict(protected_namespaces=())
 
 
-class ModelVersionResponseMetadata(WorkspaceScopedResponseMetadata):
+class ModelVersionResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for model versions."""
 
     description: Optional[str] = Field(
@@ -192,7 +192,7 @@ class ModelVersionResponseMetadata(WorkspaceScopedResponseMetadata):
     )
 
 
-class ModelVersionResponseResources(WorkspaceScopedResponseResources):
+class ModelVersionResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the model version entity."""
 
     services: Page[ServiceResponse] = Field(
@@ -201,7 +201,7 @@ class ModelVersionResponseResources(WorkspaceScopedResponseResources):
 
 
 class ModelVersionResponse(
-    WorkspaceScopedResponse[
+    ProjectScopedResponse[
         ModelVersionResponseBody,
         ModelVersionResponseMetadata,
         ModelVersionResponseResources,
@@ -566,21 +566,21 @@ class ModelVersionResponse(
 # ------------------ Filter Model ------------------
 
 
-class ModelVersionFilter(WorkspaceScopedFilter, TaggableFilter):
+class ModelVersionFilter(ProjectScopedFilter, TaggableFilter):
     """Filter model for model versions."""
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.FILTER_EXCLUDE_FIELDS,
         *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "model",
         "run_metadata",
     ]
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *ProjectScopedFilter.CUSTOM_SORTING_OPTIONS,
         *TaggableFilter.CUSTOM_SORTING_OPTIONS,
     ]
     CLI_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.CLI_EXCLUDE_FIELDS,
         *TaggableFilter.CLI_EXCLUDE_FIELDS,
         "model",
     ]

zenml/models/v2/core/pipeline.py

@@ -34,13 +34,13 @@ from zenml.constants import (
 from zenml.enums import ExecutionStatus
 from zenml.models.v2.base.base import BaseUpdate
 from zenml.models.v2.base.scoped import (
+    ProjectScopedFilter,
+    ProjectScopedRequest,
+    ProjectScopedResponse,
+    ProjectScopedResponseBody,
+    ProjectScopedResponseMetadata,
+    ProjectScopedResponseResources,
     TaggableFilter,
-    WorkspaceScopedFilter,
-    WorkspaceScopedRequest,
-    WorkspaceScopedResponse,
-    WorkspaceScopedResponseBody,
-    WorkspaceScopedResponseMetadata,
-    WorkspaceScopedResponseResources,
 )
 from zenml.models.v2.core.tag import TagResponse
 
@@ -55,7 +55,7 @@ AnyQuery = TypeVar("AnyQuery", bound=Any)
 # ------------------ Request Model ------------------
 
 
-class PipelineRequest(WorkspaceScopedRequest):
+class PipelineRequest(ProjectScopedRequest):
     """Request model for pipelines."""
 
     name: str = Field(
@@ -95,7 +95,7 @@ class PipelineUpdate(BaseUpdate):
 # ------------------ Response Model ------------------
 
 
-class PipelineResponseBody(WorkspaceScopedResponseBody):
+class PipelineResponseBody(ProjectScopedResponseBody):
     """Response body for pipelines."""
 
     latest_run_id: Optional[UUID] = Field(
@@ -108,7 +108,7 @@ class PipelineResponseBody(WorkspaceScopedResponseBody):
     )
 
 
-class PipelineResponseMetadata(WorkspaceScopedResponseMetadata):
+class PipelineResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for pipelines."""
 
     description: Optional[str] = Field(
@@ -117,7 +117,7 @@ class PipelineResponseMetadata(WorkspaceScopedResponseMetadata):
     )
 
 
-class PipelineResponseResources(WorkspaceScopedResponseResources):
+class PipelineResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the pipeline entity."""
 
     latest_run_user: Optional["UserResponse"] = Field(
@@ -130,7 +130,7 @@ class PipelineResponseResources(WorkspaceScopedResponseResources):
 
 
 class PipelineResponse(
-    WorkspaceScopedResponse[
+    ProjectScopedResponse[
         PipelineResponseBody,
         PipelineResponseMetadata,
         PipelineResponseResources,
@@ -257,21 +257,21 @@ class PipelineResponse(
 # ------------------ Filter Model ------------------
 
 
-class PipelineFilter(WorkspaceScopedFilter, TaggableFilter):
+class PipelineFilter(ProjectScopedFilter, TaggableFilter):
     """Pipeline filter model."""
 
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *ProjectScopedFilter.CUSTOM_SORTING_OPTIONS,
         *TaggableFilter.CUSTOM_SORTING_OPTIONS,
         SORT_PIPELINES_BY_LATEST_RUN_KEY,
     ]
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.FILTER_EXCLUDE_FIELDS,
         *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "latest_run_status",
     ]
     CLI_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.CLI_EXCLUDE_FIELDS,
         *TaggableFilter.CLI_EXCLUDE_FIELDS,
     ]
 

zenml/models/v2/core/pipeline_build.py

@@ -31,12 +31,12 @@ from pydantic import Field
 
 from zenml.models.v2.base.base import BaseZenModel
 from zenml.models.v2.base.scoped import (
-    WorkspaceScopedFilter,
-    WorkspaceScopedRequest,
-    WorkspaceScopedResponse,
-    WorkspaceScopedResponseBody,
-    WorkspaceScopedResponseMetadata,
-    WorkspaceScopedResponseResources,
+    ProjectScopedFilter,
+    ProjectScopedRequest,
+    ProjectScopedResponse,
+    ProjectScopedResponseBody,
+    ProjectScopedResponseMetadata,
+    ProjectScopedResponseResources,
 )
 from zenml.models.v2.misc.build_item import BuildItem
 
@@ -171,7 +171,7 @@ class PipelineBuildBase(BaseZenModel):
             )
 
 
-class PipelineBuildRequest(PipelineBuildBase, WorkspaceScopedRequest):
+class PipelineBuildRequest(PipelineBuildBase, ProjectScopedRequest):
     """Request model for pipelines builds."""
 
     checksum: Optional[str] = Field(title="The build checksum.", default=None)
@@ -193,11 +193,11 @@ class PipelineBuildRequest(PipelineBuildBase, WorkspaceScopedRequest):
 
 
 # ------------------ Response Model ------------------
-class PipelineBuildResponseBody(WorkspaceScopedResponseBody):
+class PipelineBuildResponseBody(ProjectScopedResponseBody):
     """Response body for pipeline builds."""
 
 
-class PipelineBuildResponseMetadata(WorkspaceScopedResponseMetadata):
+class PipelineBuildResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for pipeline builds."""
 
     pipeline: Optional["PipelineResponse"] = Field(
@@ -231,12 +231,12 @@ class PipelineBuildResponseMetadata(WorkspaceScopedResponseMetadata):
     )
 
 
-class PipelineBuildResponseResources(WorkspaceScopedResponseResources):
+class PipelineBuildResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the pipeline build entity."""
 
 
 class PipelineBuildResponse(
-    WorkspaceScopedResponse[
+    ProjectScopedResponse[
         PipelineBuildResponseBody,
         PipelineBuildResponseMetadata,
         PipelineBuildResponseResources,
@@ -278,7 +278,7 @@ class PipelineBuildResponse(
                 exclude={
                     "pipeline",
                     "stack",
-                    "workspace",
+                    "project",
                 }
             )
         )
@@ -473,11 +473,11 @@ class PipelineBuildResponse(
 # ------------------ Filter Model ------------------
 
 
-class PipelineBuildFilter(WorkspaceScopedFilter):
+class PipelineBuildFilter(ProjectScopedFilter):
     """Model to enable advanced filtering of all pipeline builds."""
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.FILTER_EXCLUDE_FIELDS,
         "container_registry_id",
     ]
 

zenml/models/v2/core/pipeline_deployment.py

@@ -24,12 +24,12 @@ from zenml.config.step_configurations import Step
 from zenml.models.v2.base.base import BaseZenModel
 from zenml.models.v2.base.page import Page
 from zenml.models.v2.base.scoped import (
-    WorkspaceScopedFilter,
-    WorkspaceScopedRequest,
-    WorkspaceScopedResponse,
-    WorkspaceScopedResponseBody,
-    WorkspaceScopedResponseMetadata,
-    WorkspaceScopedResponseResources,
+    ProjectScopedFilter,
+    ProjectScopedRequest,
+    ProjectScopedResponse,
+    ProjectScopedResponseBody,
+    ProjectScopedResponseMetadata,
+    ProjectScopedResponseResources,
 )
 from zenml.models.v2.core.code_reference import (
     CodeReferenceRequest,
@@ -93,9 +93,7 @@ class PipelineDeploymentBase(BaseZenModel):
         )
 
 
-class PipelineDeploymentRequest(
-    PipelineDeploymentBase, WorkspaceScopedRequest
-):
+class PipelineDeploymentRequest(PipelineDeploymentBase, ProjectScopedRequest):
     """Request model for pipeline deployments."""
 
     stack: UUID = Field(title="The stack associated with the deployment.")
@@ -129,11 +127,11 @@ class PipelineDeploymentRequest(
 # ------------------ Response Model ------------------
 
 
-class PipelineDeploymentResponseBody(WorkspaceScopedResponseBody):
+class PipelineDeploymentResponseBody(ProjectScopedResponseBody):
     """Response body for pipeline deployments."""
 
 
-class PipelineDeploymentResponseMetadata(WorkspaceScopedResponseMetadata):
+class PipelineDeploymentResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for pipeline deployments."""
 
     run_name_template: str = Field(
@@ -188,7 +186,7 @@ class PipelineDeploymentResponseMetadata(WorkspaceScopedResponseMetadata):
     )
 
 
-class PipelineDeploymentResponseResources(WorkspaceScopedResponseResources):
+class PipelineDeploymentResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the pipeline deployment entity."""
 
     triggers: TriggerPage = Field(  # type: ignore[valid-type]
@@ -197,7 +195,7 @@ class PipelineDeploymentResponseResources(WorkspaceScopedResponseResources):
 
 
 class PipelineDeploymentResponse(
-    WorkspaceScopedResponse[
+    ProjectScopedResponse[
         PipelineDeploymentResponseBody,
         PipelineDeploymentResponseMetadata,
         PipelineDeploymentResponseResources,
@@ -355,7 +353,7 @@ class PipelineDeploymentResponse(
 # ------------------ Filter Model ------------------
 
 
-class PipelineDeploymentFilter(WorkspaceScopedFilter):
+class PipelineDeploymentFilter(ProjectScopedFilter):
     """Model to enable advanced filtering of all pipeline deployments."""
 
     pipeline_id: Optional[Union[UUID, str]] = Field(

zenml/models/v2/core/pipeline_run.py

@@ -36,13 +36,13 @@ from zenml.enums import ExecutionStatus
 from zenml.metadata.metadata_types import MetadataType
 from zenml.models.v2.base.base import BaseUpdate
 from zenml.models.v2.base.scoped import (
+    ProjectScopedFilter,
+    ProjectScopedRequest,
+    ProjectScopedResponse,
+    ProjectScopedResponseBody,
+    ProjectScopedResponseMetadata,
+    ProjectScopedResponseResources,
     TaggableFilter,
-    WorkspaceScopedFilter,
-    WorkspaceScopedRequest,
-    WorkspaceScopedResponse,
-    WorkspaceScopedResponseBody,
-    WorkspaceScopedResponseMetadata,
-    WorkspaceScopedResponseResources,
 )
 from zenml.models.v2.core.model_version import ModelVersionResponse
 from zenml.models.v2.core.tag import TagResponse
@@ -71,7 +71,7 @@ AnyQuery = TypeVar("AnyQuery", bound=Any)
 # ------------------ Request Model ------------------
 
 
-class PipelineRunRequest(WorkspaceScopedRequest):
+class PipelineRunRequest(ProjectScopedRequest):
     """Request model for pipeline runs."""
 
     name: str = Field(
@@ -150,7 +150,7 @@ class PipelineRunUpdate(BaseUpdate):
 # ------------------ Response Model ------------------
 
 
-class PipelineRunResponseBody(WorkspaceScopedResponseBody):
+class PipelineRunResponseBody(ProjectScopedResponseBody):
     """Response body for pipeline runs."""
 
     status: ExecutionStatus = Field(
@@ -186,7 +186,7 @@ class PipelineRunResponseBody(WorkspaceScopedResponseBody):
     model_config = ConfigDict(protected_namespaces=())
 
 
-class PipelineRunResponseMetadata(WorkspaceScopedResponseMetadata):
+class PipelineRunResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for pipeline runs."""
 
     run_metadata: Dict[str, MetadataType] = Field(
@@ -244,7 +244,7 @@ class PipelineRunResponseMetadata(WorkspaceScopedResponseMetadata):
     )
 
 
-class PipelineRunResponseResources(WorkspaceScopedResponseResources):
+class PipelineRunResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the pipeline run entity."""
 
     model_version: Optional[ModelVersionResponse] = None
@@ -262,7 +262,7 @@ class PipelineRunResponseResources(WorkspaceScopedResponseResources):
 
 
 class PipelineRunResponse(
-    WorkspaceScopedResponse[
+    ProjectScopedResponse[
         PipelineRunResponseBody,
         PipelineRunResponseMetadata,
         PipelineRunResponseResources,
@@ -582,11 +582,11 @@ class PipelineRunResponse(
 # ------------------ Filter Model ------------------
 
 
-class PipelineRunFilter(WorkspaceScopedFilter, TaggableFilter):
-    """Model to enable advanced filtering of all Workspaces."""
+class PipelineRunFilter(ProjectScopedFilter, TaggableFilter):
+    """Model to enable advanced filtering of all pipeline runs."""
 
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *ProjectScopedFilter.CUSTOM_SORTING_OPTIONS,
         *TaggableFilter.CUSTOM_SORTING_OPTIONS,
         "tag",
         "stack",
@@ -595,7 +595,7 @@ class PipelineRunFilter(WorkspaceScopedFilter, TaggableFilter):
         "model_version",
     ]
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.FILTER_EXCLUDE_FIELDS,
         *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "unlisted",
         "code_repository_id",
@@ -613,7 +613,7 @@ class PipelineRunFilter(WorkspaceScopedFilter, TaggableFilter):
         "run_metadata",
     ]
     CLI_EXCLUDE_FIELDS = [
-        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *ProjectScopedFilter.CLI_EXCLUDE_FIELDS,
         *TaggableFilter.CLI_EXCLUDE_FIELDS,
     ]