zenml-nightly 0.75.0.dev20250317__py3-none-any.whl → 0.80.0.dev20250321__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.75.0.dev20250317
+0.80.0.dev20250321

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/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/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/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()

zenml/integrations/discord/__init__.py

@@ -44,5 +44,3 @@ class DiscordIntegration(Integration):
 
         return [DiscordAlerterFlavor]
 
-
-DiscordIntegration.check_installation()

zenml/integrations/evidently/__init__.py

@@ -87,5 +87,3 @@ class EvidentlyIntegration(Integration):
 
         return [EvidentlyDataValidatorFlavor]
 
-
-EvidentlyIntegration.check_installation()

zenml/integrations/facets/__init__.py

@@ -45,5 +45,3 @@ class FacetsIntegration(Integration):
         return cls.REQUIREMENTS + \
             PandasIntegration.get_requirements(target_os=target_os)
 
-
-FacetsIntegration.check_installation()

zenml/integrations/feast/__init__.py

@@ -60,5 +60,3 @@ class FeastIntegration(Integration):
         return cls.REQUIREMENTS + \
             PandasIntegration.get_requirements(target_os=target_os)
 
-
-FeastIntegration.check_installation()

zenml/integrations/gcp/__init__.py

@@ -84,6 +84,3 @@ class GcpIntegration(Integration):
             VertexOrchestratorFlavor,
             VertexStepOperatorFlavor,
         ]
-
-
-GcpIntegration.check_installation()

zenml/integrations/gcp/orchestrators/vertex_orchestrator.py

@@ -476,6 +476,11 @@ class VertexOrchestrator(ContainerizedOrchestrator, GoogleCredentialsMixin):
                             "Volume mounts are set but not supported in "
                             "Vertex with Kubeflow Pipelines 2.x. Ignoring..."
                         )
+                    if pod_settings.env or pod_settings.env_from:
+                        logger.warning(
+                            "Environment variables are set but not supported "
+                            "in Vertex with Vertex Pipelines 2.x. Ignoring..."
+                        )
                     for key in pod_settings.node_selectors:
                         if (
                             key

zenml/integrations/github/__init__.py

@@ -39,5 +39,3 @@ class GitHubIntegration(Integration):
 
         return [GithubWebhookEventSourceFlavor]
 
-
-GitHubIntegration.check_installation()

zenml/integrations/github/code_repositories/github_code_repository.py

@@ -20,6 +20,7 @@ from uuid import uuid4
 
 import requests
 from github import Consts, Github, GithubException
+from github.Auth import Token
 from github.Repository import Repository
 
 from zenml.code_repositories import (
@@ -145,9 +146,10 @@ class GitHubCodeRepository(BaseCodeRepository):
         """
         try:
             self._github_session = Github(
-                login_or_token=self.config.token,
                 base_url=self.config.api_url or Consts.DEFAULT_BASE_URL,
+                auth=Token(self.config.token) if self.config.token else None,
             )
+
             if self.config.token:
                 user = self._github_session.get_user().login
                 logger.debug(f"Logged in as {user}")
@@ -235,12 +237,17 @@ class GitHubCodeRepository(BaseCodeRepository):
         Returns:
             Whether the remote url is correct.
         """
-        https_url = f"https://{self.config.host}/{self.config.owner}/{self.config.repository}.git"
+        host = self.config.host or "github.com"
+        host = host.rstrip("/")
+
+        https_url = (
+            f"https://{host}/{self.config.owner}/{self.config.repository}.git"
+        )
         if url == https_url:
             return True
 
         ssh_regex = re.compile(
-            f".*@{self.config.host}:{self.config.owner}/{self.config.repository}.git"
+            f".*@{host}:{self.config.owner}/{self.config.repository}.git"
         )
         if ssh_regex.fullmatch(url):
             return True

zenml/integrations/gitlab/__init__.py

@@ -25,4 +25,3 @@ class GitLabIntegration(Integration):
     REQUIREMENTS: List[str] = ["python-gitlab"]
 
 
-GitLabIntegration.check_installation()

zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py

@@ -176,10 +176,13 @@ class GitLabCodeRepository(BaseCodeRepository):
         Returns:
             Whether the remote url is correct.
         """
+        host = self.config.host or "gitlab.com"
+        host = host.rstrip("/")
+
         parsed_url = urlparse(url)
         if (
             parsed_url.scheme == "https"
-            and parsed_url.hostname == self.config.host
+            and parsed_url.hostname == host
             and parsed_url.path
             == f"/{self.config.group}/{self.config.project}.git"
         ):
@@ -188,7 +191,7 @@ class GitLabCodeRepository(BaseCodeRepository):
         ssh_regex = re.compile(
             r"^(?P<scheme_with_delimiter>ssh://)?"
             r"(?P<userinfo>git)"
-            f"@{self.config.host}:"
+            f"@{host}:"
             r"(?P<port>\d+)?"
             r"(?(scheme_with_delimiter)/|/?)"
             f"{self.config.group}/{self.config.project}.git$",

zenml/integrations/great_expectations/__init__.py

@@ -66,6 +66,3 @@ class GreatExpectationsIntegration(Integration):
 
         return cls.REQUIREMENTS + \
             PandasIntegration.get_requirements(target_os=target_os)
-
-
-GreatExpectationsIntegration.check_installation()