zenml-nightly 0.75.0.dev20250318__py3-none-any.whl → 0.80.0.dev20250322__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.75.0.dev20250318
+0.80.0.dev20250322

zenml/__init__.py

@@ -31,7 +31,7 @@ init_logging()
 from zenml.models import *  # noqa: F401
 
 # Define public Python API
-from zenml.zen_server.utils import show_dashboard as show
+from zenml.utils.dashboard_utils import show_dashboard as show
 from zenml.artifacts.utils import (
     log_artifact_metadata,
     save_artifact,
@@ -54,6 +54,8 @@ from zenml.entrypoints import entrypoint
 
 __all__ = [
     "add_tags",
+    "remove_tags",
+    "Tag",
     "ArtifactConfig",
     "ExternalArtifact",
     "get_pipeline_context",

zenml/artifacts/artifact_config.py

@@ -107,7 +107,7 @@ class ArtifactConfig(BaseModel):
                 "`ArtifactConfig(..., is_model_artifact=True)` is deprecated "
                 "and will be removed soon. Use `ArtifactConfig(..., "
                 "artifact_type=ArtifactType.MODEL)` instead. For more info: "
-                "https://docs.zenml.io/user-guide/starter-guide/manage-artifacts"
+                "https://docs.zenml.io/user-guides/starter-guide/manage-artifacts"
             )
             data.setdefault("artifact_type", ArtifactType.MODEL)
         elif is_deployment_artifact:
@@ -115,7 +115,7 @@ class ArtifactConfig(BaseModel):
                 "`ArtifactConfig(..., is_deployment_artifact=True)` is "
                 "deprecated and will be removed soon. Use `ArtifactConfig(..., "
                 "artifact_type=ArtifactType.SERVICE)` instead. For more info: "
-                "https://docs.zenml.io/user-guide/starter-guide/manage-artifacts"
+                "https://docs.zenml.io/user-guides/starter-guide/manage-artifacts"
             )
             data.setdefault("artifact_type", ArtifactType.SERVICE)
 

zenml/artifacts/utils.py

@@ -35,9 +35,7 @@ from zenml.artifacts.preexisting_data_materializer import (
     PreexistingDataMaterializer,
 )
 from zenml.client import Client
-from zenml.constants import (
-    MODEL_METADATA_YAML_FILE_NAME,
-)
+from zenml.constants import MODEL_METADATA_YAML_FILE_NAME
 from zenml.enums import (
     ArtifactSaveType,
     ArtifactType,
@@ -45,10 +43,7 @@ from zenml.enums import (
     StackComponentType,
     VisualizationType,
 )
-from zenml.exceptions import (
-    DoesNotExistException,
-    StepContextError,
-)
+from zenml.exceptions import DoesNotExistException, StepContextError
 from zenml.io import fileio
 from zenml.logger import get_logger
 from zenml.metadata.metadata_types import validate_metadata
@@ -77,6 +72,7 @@ if TYPE_CHECKING:
 
 logger = get_logger(__name__)
 
+
 # ----------
 # Public API
 # ----------
@@ -320,7 +316,7 @@ def register_artifact(
         The saved artifact response.
 
     Raises:
-        FileNotFoundError: If the folder URI is outside of the artifact store
+        FileNotFoundError: If the folder URI is outside the artifact store
             bounds.
     """
     client = Client()

zenml/cli/__init__.py

@@ -80,7 +80,7 @@ You can also pass in a directory path manually using the
 zenml init --path /path/to/dir
 ```
 
-If you wish to use one of [the available ZenML project templates](https://docs.zenml.io/how-to/setting-up-a-project-repository/using-project-templates#list-of-zenml-project-templates)
+If you wish to use one of [the available ZenML project templates](https://docs.zenml.io/how-to/project-setup-and-management/collaborate-with-team/project-templates)
 to generate a ready-to-use project scaffold in your repository, you can do so by
 passing the ``--template`` option:
 
@@ -1389,7 +1389,7 @@ zenml model-registry --help
 Managing your Stacks
 --------------------
 
-[The stack](https://docs.zenml.io/user-guide/production-guide/understand-stacks)
+[The stack](https://docs.zenml.io/user-guides/production-guide/understand-stacks)
 is a grouping of your artifact store, your orchestrator, and other
 optional MLOps tools like experiment trackers or model deployers.
 With the ZenML tool, switching from a local stack to a distributed cloud
@@ -1588,7 +1588,7 @@ Managing your Models
 --------------------
 
 ZenML provides several CLI commands to help you administer your models and
-their versions as part of [the Model Control Plane](https://docs.zenml.io/user-guide/starter-guide/track-ml-models).
+their versions as part of [the Model Control Plane](https://docs.zenml.io/user-guides/starter-guide/track-ml-models).
 
 To register a new model, you can use the following CLI command:
 
@@ -1672,7 +1672,7 @@ Managing your Pipelines & Artifacts
 -----------------------------------
 
 ZenML provides several CLI commands to help you [administer your pipelines and
-pipeline runs](https://docs.zenml.io/user-guide/starter-guide/manage-artifacts).
+pipeline runs](https://docs.zenml.io/user-guides/starter-guide/manage-artifacts).
 
 To explicitly register a pipeline you need to point to a pipeline instance
 in your Python code. Let's say you have a Python file called `run.py` and
@@ -1880,7 +1880,7 @@ Connecting to a ZenML Server
 ----------------------------
 
 The ZenML client can be [configured to connect to a local ZenML server, a remote
-database or a remote ZenML server](https://docs.zenml.io/how-to/connecting-to-zenml)
+database or a remote ZenML server](https://docs.zenml.io/how-to/manage-zenml-server/connecting-to-zenml)
 with the `zenml login` command.
 
 To connect or re-connect to any ZenML server, if you know its URL, you can
@@ -2306,7 +2306,7 @@ zenml service-account api-key <SERVICE_ACCOUNT_NAME> delete <API_KEY_NAME>
 Managing Code Repositories
 --------------------------
 
-[Code repositories](https://docs.zenml.io/user-guide/production-guide/connect-code-repository)
+[Code repositories](https://docs.zenml.io/user-guides/production-guide/connect-code-repository)
 enable ZenML to keep track of the code version that you use for your pipeline
 runs. Additionally, running a pipeline which is tracked in a registered code
 repository can decrease the time it takes Docker to build images for

zenml/cli/base.py

@@ -49,8 +49,8 @@ from zenml.integrations.registry import integration_registry
 from zenml.io import fileio
 from zenml.logger import get_logger
 from zenml.utils.io_utils import copy_dir, get_global_config_directory
+from zenml.utils.server_utils import get_local_server
 from zenml.utils.yaml_utils import write_yaml
-from zenml.zen_server.utils import get_local_server
 
 logger = get_logger(__name__)
 # WT_SESSION is a Windows Terminal specific environment variable. If it
@@ -264,7 +264,7 @@ def init(
         f"will only take effect when you're running ZenML from the initialized "
         f"repository root, or from a subdirectory. For more information on "
         f"repositories and configurations, please visit "
-        f"https://docs.zenml.io/user-guide/production-guide/understand-stacks."
+        f"https://docs.zenml.io/user-guides/production-guide/understand-stacks."
     )
 
 

zenml/cli/login.py

@@ -37,10 +37,10 @@ from zenml.logger import get_logger
 from zenml.login.credentials import ServerType
 from zenml.login.pro.constants import ZENML_PRO_API_URL
 from zenml.login.web_login import web_login
-from zenml.zen_server.utils import (
+from zenml.utils.dashboard_utils import show_dashboard
+from zenml.utils.server_utils import (
     connected_to_local_server,
     get_local_server,
-    show_dashboard,
 )
 
 logger = get_logger(__name__)

zenml/cli/project.py

@@ -38,7 +38,7 @@ def project() -> None:
 @project.command("list")
 @list_options(ProjectFilter)
 @click.pass_context
-def list_projects(ctx: click.Context, **kwargs: Any) -> None:
+def list_projects(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all projects.
 
     Args:

zenml/cli/server.py

@@ -27,12 +27,14 @@ from zenml.cli.login import login, logout
 from zenml.client import Client
 from zenml.config.global_config import GlobalConfiguration
 from zenml.console import console
-from zenml.enums import CliCategories, StoreType
+from zenml.enums import CliCategories, ServiceState, StoreType
 from zenml.exceptions import AuthorizationException
 from zenml.logger import get_logger
 from zenml.login.credentials import ServerCredentials, ServerType
-from zenml.services.service_status import ServiceState
-from zenml.zen_server.utils import connected_to_local_server, get_local_server
+from zenml.utils.server_utils import (
+    connected_to_local_server,
+    get_local_server,
+)
 
 logger = get_logger(__name__)
 
@@ -402,7 +404,7 @@ def connect(
             "filesystem and is no longer supported. The web login workflow will "
             "be used instead. An alternative for non-interactive environments "
             "is to create and use a service account API key (see "
-            "https://docs.zenml.io/how-to/connecting-to-zenml/connect-with-a-service-account "
+            "https://docs.zenml.io/how-to/manage-zenml-server/connecting-to-zenml/connect-with-a-service-account "
             "for more information)."
         )
 

zenml/cli/service_accounts.py

@@ -187,7 +187,7 @@ def describe_service_account(service_account_name_or_id: str) -> None:
 @service_account.command("list")
 @list_options(ServiceAccountFilter)
 @click.pass_context
-def list_service_accounts(ctx: click.Context, **kwargs: Any) -> None:
+def list_service_accounts(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all users.
 
     Args:
@@ -384,7 +384,7 @@ def describe_api_key(service_account_name_or_id: str, name_or_id: str) -> None:
 @api_key.command("list", help="List all API keys.")
 @list_options(APIKeyFilter)
 @click.pass_obj
-def list_api_keys(service_account_name_or_id: str, **kwargs: Any) -> None:
+def list_api_keys(service_account_name_or_id: str, /, **kwargs: Any) -> None:
     """List all API keys.
 
     Args:

zenml/cli/service_connectors.py

@@ -982,7 +982,7 @@ def register_service_connector(
 )
 @click.pass_context
 def list_service_connectors(
-    ctx: click.Context, labels: Optional[List[str]] = None, **kwargs: Any
+    ctx: click.Context, /, labels: Optional[List[str]] = None, **kwargs: Any
 ) -> None:
     """List all service connectors.
 

zenml/cli/stack.py

@@ -937,7 +937,7 @@ def rename_stack(
 @stack.command("list")
 @list_options(StackFilter)
 @click.pass_context
-def list_stacks(ctx: click.Context, **kwargs: Any) -> None:
+def list_stacks(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all stacks that fulfill the filter requirements.
 
     Args:

zenml/cli/stack_components.py

@@ -155,7 +155,7 @@ def generate_stack_component_list_command(
     @list_options(ComponentFilter)
     @click.pass_context
     def list_stack_components_command(
-        ctx: click.Context, **kwargs: Any
+        ctx: click.Context, /, **kwargs: Any
     ) -> None:
         """Prints a table of stack components.
 

zenml/cli/user_management.py

@@ -79,7 +79,7 @@ def describe_user(user_name_or_id: Optional[str] = None) -> None:
 @user.command("list")
 @list_options(UserFilter)
 @click.pass_context
-def list_users(ctx: click.Context, **kwargs: Any) -> None:
+def list_users(ctx: click.Context, /, **kwargs: Any) -> None:
     """List all users.
 
     Args:

zenml/cli/utils.py

@@ -58,7 +58,7 @@ from zenml.constants import (
     FILTERING_DATETIME_FORMAT,
     IS_DEBUG_ENV,
 )
-from zenml.enums import GenericFilterOps, StackComponentType
+from zenml.enums import GenericFilterOps, ServiceState, StackComponentType
 from zenml.logger import get_logger
 from zenml.model_registries.base_model_registry import (
     RegisteredModel,
@@ -75,7 +75,7 @@ from zenml.models import (
     UUIDFilter,
 )
 from zenml.models.v2.base.filter import FilterGenerator
-from zenml.services import BaseService, ServiceState
+from zenml.services import BaseService
 from zenml.stack import StackComponent
 from zenml.stack.flavor import Flavor
 from zenml.stack.stack_component import StackComponentConfig
@@ -1199,7 +1199,7 @@ def get_service_state_emoji(state: "ServiceState") -> str:
     Returns:
         String representing the emoji.
     """
-    from zenml.services.service_status import ServiceState
+    from zenml.enums import ServiceState
 
     if state == ServiceState.ACTIVE:
         return ":white_check_mark:"

zenml/client.py

@@ -65,6 +65,7 @@ from zenml.enums import (
     OAuthDeviceStatus,
     PluginSubType,
     PluginType,
+    ServiceState,
     SorterOps,
     StackComponentType,
     StoreType,
@@ -166,6 +167,7 @@ from zenml.models import (
     ServiceFilter,
     ServiceRequest,
     ServiceResponse,
+    ServiceType,
     ServiceUpdate,
     StackFilter,
     StackRequest,
@@ -173,6 +175,7 @@ from zenml.models import (
     StackUpdate,
     StepRunFilter,
     StepRunResponse,
+    StepRunUpdate,
     TagFilter,
     TagRequest,
     TagResource,
@@ -190,10 +193,6 @@ from zenml.models import (
     UserResponse,
     UserUpdate,
 )
-from zenml.models.v2.core.step_run import StepRunUpdate
-from zenml.services.service import ServiceConfig
-from zenml.services.service_status import ServiceState
-from zenml.services.service_type import ServiceType
 from zenml.utils import io_utils, source_utils
 from zenml.utils.dict_utils import dict_to_bytes
 from zenml.utils.filesync_model import FileSyncModel
@@ -203,6 +202,7 @@ from zenml.utils.uuid_utils import is_valid_uuid
 if TYPE_CHECKING:
     from zenml.metadata.metadata_types import MetadataType, MetadataTypeEnum
     from zenml.service_connectors.service_connector import ServiceConnector
+    from zenml.services.service import ServiceConfig
     from zenml.stack import Stack
     from zenml.zen_stores.base_zen_store import BaseZenStore
 
@@ -1625,7 +1625,7 @@ class Client(metaclass=ClientMetaClass):
 
     def create_service(
         self,
-        config: ServiceConfig,
+        config: "ServiceConfig",
         service_type: ServiceType,
         model_version_id: Optional[UUID] = None,
     ) -> ServiceResponse:
@@ -3896,7 +3896,7 @@ class Client(metaclass=ClientMetaClass):
         tag: Optional[str] = None,
         tags: Optional[List[str]] = None,
         user: Optional[Union[UUID, str]] = None,
-        run_metadata: Optional[Dict[str, Any]] = None,
+        run_metadata: Optional[List[str]] = None,
         pipeline: Optional[Union[UUID, str]] = None,
         code_repository: Optional[Union[UUID, str]] = None,
         model: Optional[Union[UUID, str]] = None,
@@ -4049,7 +4049,7 @@ class Client(metaclass=ClientMetaClass):
         user: Optional[Union[UUID, str]] = None,
         model_version_id: Optional[Union[str, UUID]] = None,
         model: Optional[Union[UUID, str]] = None,
-        run_metadata: Optional[Dict[str, Any]] = None,
+        run_metadata: Optional[List[str]] = None,
         hydrate: bool = False,
     ) -> Page[StepRunResponse]:
         """List all pipelines.
@@ -4362,7 +4362,7 @@ class Client(metaclass=ClientMetaClass):
         user: Optional[Union[UUID, str]] = None,
         model: Optional[Union[UUID, str]] = None,
         pipeline_run: Optional[Union[UUID, str]] = None,
-        run_metadata: Optional[Dict[str, Any]] = None,
+        run_metadata: Optional[List[str]] = None,
         tag: Optional[str] = None,
         tags: Optional[List[str]] = None,
         hydrate: bool = False,
@@ -6509,7 +6509,7 @@ class Client(metaclass=ClientMetaClass):
         id: Optional[Union[UUID, str]] = None,
         number: Optional[int] = None,
         stage: Optional[Union[str, ModelStages]] = None,
-        run_metadata: Optional[Dict[str, str]] = None,
+        run_metadata: Optional[List[str]] = None,
         user: Optional[Union[UUID, str]] = None,
         hydrate: bool = False,
         tag: Optional[str] = None,

zenml/config/compiler.py

@@ -599,7 +599,7 @@ class Compiler:
                 f"no steps. Please make sure that your steps are decorated "
                 "with `@step` and that at least one step is called within the "
                 "pipeline. For more information, see "
-                "https://docs.zenml.io/user-guide/starter-guide."
+                "https://docs.zenml.io/user-guides/starter-guide."
             )
 
         additional_spec_args: Dict[str, Any] = {

zenml/config/server_config.py

@@ -598,7 +598,7 @@ class ServerConfiguration(BaseModel):
             )
             server_config.feature_gate_implementation_source = "zenml.zen_server.feature_gate.zenml_cloud_feature_gate.ZenMLCloudFeatureGateInterface"
             server_config.reportable_resources = DEFAULT_REPORTABLE_RESOURCES
-            server_config.dashboard_url = f"{server_pro_config.dashboard_url}/workspaces/{server_pro_config.workspace_id}"
+            server_config.dashboard_url = f"{server_pro_config.dashboard_url}/workspaces/{server_pro_config.workspace_name}"
             server_config.metadata.update(
                 dict(
                     account_id=str(server_pro_config.organization_id),

zenml/constants.py

@@ -329,7 +329,7 @@ DEFAULT_ZENML_SERVER_SECURE_HEADERS_REPORT_TO = "default"
 DEFAULT_ZENML_SERVER_REPORT_USER_ACTIVITY_TO_DB_SECONDS = 30
 DEFAULT_ZENML_SERVER_MAX_REQUEST_BODY_SIZE_IN_BYTES = 256 * 1024 * 1024
 
-DEFAULT_REPORTABLE_RESOURCES = ["pipeline", "pipeline_run", "model"]
+DEFAULT_REPORTABLE_RESOURCES = ["project", "pipeline", "pipeline_run", "model"]
 
 # API Endpoint paths:
 ACTIVATE = "/activate"
@@ -368,6 +368,7 @@ PIPELINE_DEPLOYMENTS = "/pipeline_deployments"
 PIPELINES = "/pipelines"
 PIPELINE_SPEC = "/pipeline-spec"
 PLUGIN_FLAVORS = "/plugin-flavors"
+PROJECTS = "/projects"
 REFRESH = "/refresh"
 RUNS = "/runs"
 RUN_TEMPLATES = "/run_templates"

zenml/enums.py

@@ -405,6 +405,7 @@ class OnboardingStep(StrEnum):
     """All onboarding steps."""
 
     DEVICE_VERIFIED = "device_verified"
+    PROJECT_CREATED = "project_created"
     PIPELINE_RUN = "pipeline_run"
     STARTER_SETUP_COMPLETED = "starter_setup_completed"
     STACK_WITH_REMOTE_ORCHESTRATOR_CREATED = (
@@ -422,3 +423,14 @@ class StackDeploymentProvider(StrEnum):
     AWS = "aws"
     GCP = "gcp"
     AZURE = "azure"
+
+
+class ServiceState(StrEnum):
+    """Possible states for the service and service endpoint."""
+
+    INACTIVE = "inactive"
+    ACTIVE = "active"
+    PENDING_STARTUP = "pending_startup"
+    PENDING_SHUTDOWN = "pending_shutdown"
+    ERROR = "error"
+    SCALED_TO_ZERO = "scaled_to_zero"

zenml/integrations/__init__.py

@@ -17,67 +17,3 @@ The ZenML integrations module contains sub-modules for each integration that we
 support. This includes orchestrators like Apache Airflow, visualization tools
 like the ``facets`` library, as well as deep learning libraries like PyTorch.
 """
-from zenml.integrations.airflow import AirflowIntegration  # noqa
-from zenml.integrations.argilla import ArgillaIntegration  # noqa
-from zenml.integrations.aws import AWSIntegration  # noqa
-from zenml.integrations.azure import AzureIntegration  # noqa
-from zenml.integrations.bentoml import BentoMLIntegration  # noqa
-from zenml.integrations.bitbucket import BitbucketIntegration  # noqa
-from zenml.integrations.databricks import DatabricksIntegration  # noqa
-from zenml.integrations.comet import CometIntegration  # noqa
-from zenml.integrations.deepchecks import DeepchecksIntegration  # noqa
-from zenml.integrations.discord import DiscordIntegration  # noqa
-from zenml.integrations.evidently import EvidentlyIntegration  # noqa
-from zenml.integrations.facets import FacetsIntegration  # noqa
-from zenml.integrations.feast import FeastIntegration  # noqa
-from zenml.integrations.gcp import GcpIntegration  # noqa
-from zenml.integrations.github import GitHubIntegration  # noqa
-from zenml.integrations.gitlab import GitLabIntegration  # noqa
-from zenml.integrations.great_expectations import (  # noqa
-    GreatExpectationsIntegration,
-)
-from zenml.integrations.lightning import LightningIntegration  # noqa
-from zenml.integrations.huggingface import HuggingfaceIntegration  # noqa
-from zenml.integrations.hyperai import HyperAIIntegration  # noqa
-from zenml.integrations.kaniko import KanikoIntegration  # noqa
-from zenml.integrations.kubeflow import KubeflowIntegration  # noqa
-from zenml.integrations.kubernetes import KubernetesIntegration  # noqa
-from zenml.integrations.label_studio import LabelStudioIntegration  # noqa
-from zenml.integrations.langchain import LangchainIntegration  # noqa
-from zenml.integrations.lightgbm import LightGBMIntegration  # noqa
-
-# from zenml.integrations.llama_index import LlamaIndexIntegration  # noqa
-from zenml.integrations.mlflow import MlflowIntegration  # noqa
-from zenml.integrations.modal import ModalIntegration  # noqa
-from zenml.integrations.neptune import NeptuneIntegration  # noqa
-from zenml.integrations.neural_prophet import NeuralProphetIntegration  # noqa
-from zenml.integrations.numpy import NumpyIntegration  # noqa
-from zenml.integrations.openai import OpenAIIntegration  # noqa
-from zenml.integrations.pandas import PandasIntegration  # noqa
-from zenml.integrations.pigeon import PigeonIntegration  # noqa
-from zenml.integrations.pillow import PillowIntegration  # noqa
-from zenml.integrations.polars import PolarsIntegration  # noqa
-from zenml.integrations.prodigy import ProdigyIntegration  # noqa
-from zenml.integrations.pycaret import PyCaretIntegration  # noqa
-from zenml.integrations.pytorch import PytorchIntegration  # noqa
-from zenml.integrations.pytorch_lightning import (  # noqa
-    PytorchLightningIntegration,
-)
-from zenml.integrations.s3 import S3Integration  # noqa
-from zenml.integrations.scipy import ScipyIntegration  # noqa
-from zenml.integrations.seldon import SeldonIntegration  # noqa
-from zenml.integrations.sklearn import SklearnIntegration  # noqa
-from zenml.integrations.skypilot_aws import SkypilotAWSIntegration  # noqa
-from zenml.integrations.skypilot_gcp import SkypilotGCPIntegration  # noqa
-from zenml.integrations.skypilot_azure import SkypilotAzureIntegration  # noqa
-from zenml.integrations.skypilot_lambda import SkypilotLambdaIntegration  # noqa
-from zenml.integrations.skypilot_kubernetes import SkypilotKubernetesIntegration  # noqa
-from zenml.integrations.slack import SlackIntegration  # noqa
-from zenml.integrations.spark import SparkIntegration  # noqa
-from zenml.integrations.tekton import TektonIntegration  # noqa
-from zenml.integrations.tensorboard import TensorBoardIntegration  # noqa
-from zenml.integrations.tensorflow import TensorflowIntegration  # noqa
-from zenml.integrations.wandb import WandbIntegration  # noqa
-from zenml.integrations.whylogs import WhylogsIntegration  # noqa
-from zenml.integrations.xgboost import XgboostIntegration  # noqa
-from zenml.integrations.vllm import VLLMIntegration  # noqa

zenml/integrations/airflow/__init__.py

@@ -43,5 +43,3 @@ class AirflowIntegration(Integration):
 
         return [AirflowOrchestratorFlavor]
 
-
-AirflowIntegration.check_installation()

zenml/integrations/argilla/__init__.py

@@ -43,4 +43,3 @@ class ArgillaIntegration(Integration):
         return [ArgillaAnnotatorFlavor]
 
 
-ArgillaIntegration.check_installation()

zenml/integrations/aws/__init__.py

@@ -72,6 +72,3 @@ class AWSIntegration(Integration):
             SagemakerStepOperatorFlavor,
             SagemakerOrchestratorFlavor,
         ]
-
-
-AWSIntegration.check_installation()

zenml/integrations/azure/__init__.py

@@ -80,5 +80,3 @@ class AzureIntegration(Integration):
             AzureMLOrchestratorFlavor,
         ]
 
-
-AzureIntegration.check_installation()

zenml/integrations/azure/service_connectors/azure_service_connector.py

@@ -1738,7 +1738,7 @@ class AzureServiceConnector(ServiceConnector):
             ):
                 assert isinstance(self.config, AzureServicePrincipalConfig)
                 username = str(self.config.client_id)
-                password = str(self.config.client_secret)
+                password = self.config.client_secret.get_secret_value()
 
             # Without a service principal, we try to use the AzureDefaultCredentials to authenticate against the ACR.
             # If this fails, we try to use the admin account.

zenml/integrations/bentoml/__init__.py

@@ -53,5 +53,3 @@ class BentoMLIntegration(Integration):
 
         return [BentoMLModelDeployerFlavor]
 
-
-BentoMLIntegration.check_installation()

zenml/integrations/bentoml/services/bentoml_container_deployment.py

@@ -23,12 +23,14 @@ from bentoml.client import Client
 
 from zenml.client import Client as ZenMLClient
 from zenml.constants import DEFAULT_LOCAL_SERVICE_IP_ADDRESS
+from zenml.enums import ServiceState
 from zenml.integrations.bentoml.constants import (
     BENTOML_DEFAULT_PORT,
     BENTOML_HEALTHCHECK_URL_PATH,
     BENTOML_PREDICTION_URL_PATH,
 )
 from zenml.logger import get_logger
+from zenml.models.v2.misc.service import ServiceType
 from zenml.services.container.container_service import (
     ContainerService,
     ContainerServiceConfig,
@@ -43,8 +45,6 @@ from zenml.services.service_monitor import (
     HTTPEndpointHealthMonitor,
     HTTPEndpointHealthMonitorConfig,
 )
-from zenml.services.service_status import ServiceState
-from zenml.services.service_type import ServiceType
 
 logger = get_logger(__name__)
 

zenml/integrations/bentoml/services/bentoml_local_deployment.py

@@ -26,6 +26,7 @@ from zenml.integrations.bentoml.constants import (
     BENTOML_PREDICTION_URL_PATH,
 )
 from zenml.logger import get_logger
+from zenml.models.v2.misc.service import ServiceType
 from zenml.services import (
     HTTPEndpointHealthMonitor,
     HTTPEndpointHealthMonitorConfig,
@@ -34,7 +35,6 @@ from zenml.services import (
     LocalDaemonServiceEndpoint,
     LocalDaemonServiceEndpointConfig,
     ServiceEndpointProtocol,
-    ServiceType,
 )
 from zenml.services.service import BaseDeploymentService
 

zenml/integrations/bentoml/steps/bentoml_deployer.py

@@ -35,8 +35,8 @@ from zenml.integrations.bentoml.services.deployment_type import (
     BentoMLDeploymentType,
 )
 from zenml.logger import get_logger
+from zenml.models.v2.misc.service import ServiceType
 from zenml.services.service import BaseService, ServiceConfig
-from zenml.services.service_type import ServiceType
 from zenml.utils import source_utils
 
 logger = get_logger(__name__)

zenml/integrations/bitbucket/__init__.py

@@ -38,5 +38,3 @@ class BitbucketIntegration(Integration):
 
         return [BitbucketWebhookEventSourceFlavor]
 
-
-BitbucketIntegration.check_installation()

zenml/integrations/comet/__init__.py

@@ -45,5 +45,3 @@ class CometIntegration(Integration):
 
         return [CometExperimentTrackerFlavor]
 
-
-CometIntegration.check_installation()

zenml/integrations/databricks/__init__.py

@@ -66,5 +66,3 @@ class DatabricksIntegration(Integration):
             DatabricksOrchestratorFlavor,
             DatabricksModelDeployerFlavor,
         ]
-
-DatabricksIntegration.check_installation()

zenml/integrations/databricks/services/databricks_deployment.py

@@ -31,6 +31,7 @@ from databricks.sdk.service.serving import (
 from pydantic import Field
 
 from zenml.client import Client
+from zenml.enums import ServiceState
 from zenml.integrations.databricks.flavors.databricks_model_deployer_flavor import (
     DatabricksBaseConfig,
 )
@@ -38,7 +39,8 @@ from zenml.integrations.databricks.utils.databricks_utils import (
     sanitize_labels,
 )
 from zenml.logger import get_logger
-from zenml.services import ServiceState, ServiceStatus, ServiceType
+from zenml.models.v2.misc.service import ServiceType
+from zenml.services import ServiceStatus
 from zenml.services.service import BaseDeploymentService, ServiceConfig
 
 logger = get_logger(__name__)

zenml/integrations/deepchecks/__init__.py

@@ -85,5 +85,3 @@ class DeepchecksIntegration(Integration):
 
         return [DeepchecksDataValidatorFlavor]
 
-
-DeepchecksIntegration.check_installation()