zenml-nightly 0.58.2.dev20240617__py3-none-any.whl → 0.58.2.dev20240619__py3-none-any.whl

zenml/integrations/kubeflow/utils.py

@@ -1,95 +0,0 @@
-#  Copyright (c) ZenML GmbH 2022. All Rights Reserved.
-#
-#  Licensed under the Apache License, Version 2.0 (the "License");
-#  you may not use this file except in compliance with the License.
-#  You may obtain a copy of the License at:
-#
-#       https://www.apache.org/licenses/LICENSE-2.0
-#
-#  Unless required by applicable law or agreed to in writing, software
-#  distributed under the License is distributed on an "AS IS" BASIS,
-#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
-#  or implied. See the License for the specific language governing
-#  permissions and limitations under the License.
-"""KFP utilities."""
-
-from typing import TYPE_CHECKING
-
-from zenml.integrations.kubernetes import serialization_utils
-from zenml.integrations.kubernetes.pod_settings import KubernetesPodSettings
-from zenml.logger import get_logger
-
-logger = get_logger(__name__)
-
-if TYPE_CHECKING:
-    from kfp.dsl import ContainerOp
-
-
-def apply_pod_settings(
-    container_op: "ContainerOp",
-    settings: KubernetesPodSettings,
-) -> None:
-    """Applies Kubernetes Pod settings to a KFP container.
-
-    Args:
-        container_op: The container to which to apply the settings.
-        settings: The settings to apply.
-    """
-    from kubernetes.client.models import (
-        V1Affinity,
-        V1Toleration,
-        V1Volume,
-        V1VolumeMount,
-    )
-
-    if settings.host_ipc:
-        logger.warning(
-            "Host IPC is set to `True` but not supported in this orchestrator. "
-            "Ignoring..."
-        )
-
-    for key, value in settings.node_selectors.items():
-        container_op.add_node_selector_constraint(label_name=key, value=value)
-
-    if settings.affinity:
-        affinity: V1Affinity = (
-            serialization_utils.deserialize_kubernetes_model(
-                settings.affinity, "V1Affinity"
-            )
-        )
-        container_op.add_affinity(affinity)
-
-    for toleration_dict in settings.tolerations:
-        toleration: V1Toleration = (
-            serialization_utils.deserialize_kubernetes_model(
-                toleration_dict, "V1Toleration"
-            )
-        )
-        container_op.add_toleration(toleration)
-
-    if settings.volumes:
-        for v in settings.volumes:
-            volume: V1Volume = (
-                serialization_utils.deserialize_kubernetes_model(v, "V1Volume")
-            )
-            container_op.add_volume(volume)
-
-    if settings.volume_mounts:
-        for v in settings.volume_mounts:
-            volume_mount: V1VolumeMount = (
-                serialization_utils.deserialize_kubernetes_model(
-                    v, "V1VolumeMount"
-                )
-            )
-            container_op.container.add_volume_mount(volume_mount)
-
-    resource_requests = settings.resources.get("requests") or {}
-    for name, value in resource_requests.items():
-        container_op.add_resource_request(name, value)
-
-    resource_limits = settings.resources.get("limits") or {}
-    for name, value in resource_limits.items():
-        container_op.add_resource_limit(name, value)
-
-    for name, value in settings.annotations.items():
-        container_op.add_pod_annotation(name, value)

zenml/models/v2/base/internal.py

@@ -1,37 +0,0 @@
-#  Copyright (c) ZenML GmbH 2023. All Rights Reserved.
-#
-#  Licensed under the Apache License, Version 2.0 (the "License");
-#  you may not use this file except in compliance with the License.
-#  You may obtain a copy of the License at:
-#
-#       https://www.apache.org/licenses/LICENSE-2.0
-#
-#  Unless required by applicable law or agreed to in writing, software
-#  distributed under the License is distributed on an "AS IS" BASIS,
-#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
-#  or implied. See the License for the specific language governing
-#  permissions and limitations under the License.
-"""Utility methods for internal models."""
-
-from typing import Type, TypeVar
-
-from zenml.models.v2.base.base import BaseRequest
-
-T = TypeVar("T", bound="BaseRequest")
-
-
-def server_owned_request_model(_cls: Type[T]) -> Type[T]:
-    """Convert a request model to a model which does not require a user ID.
-
-    Args:
-        _cls: The class to decorate
-
-    Returns:
-        The decorated class.
-    """
-    if user_field := _cls.__fields__.get("user", None):
-        user_field.required = False
-        user_field.allow_none = True
-        user_field.default = None
-
-    return _cls

zenml/models/v2/base/update.py

@@ -1,44 +0,0 @@
-#  Copyright (c) ZenML GmbH 2023. All Rights Reserved.
-#
-#  Licensed under the Apache License, Version 2.0 (the "License");
-#  you may not use this file except in compliance with the License.
-#  You may obtain a copy of the License at:
-#
-#       https://www.apache.org/licenses/LICENSE-2.0
-#
-#  Unless required by applicable law or agreed to in writing, software
-#  distributed under the License is distributed on an "AS IS" BASIS,
-#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
-#  or implied. See the License for the specific language governing
-#  permissions and limitations under the License.
-"""Utility methods for base models."""
-
-from typing import TYPE_CHECKING, Type, TypeVar
-
-from pydantic.config import Extra
-
-if TYPE_CHECKING:
-    from zenml.models.v2.base.base import BaseRequest
-
-    T = TypeVar("T", bound=BaseRequest)
-
-
-def update_model(_cls: Type["T"]) -> Type["T"]:
-    """Base update model.
-
-    This is used as a decorator on top of request models to convert them
-    into update models where the fields are optional and can be set to None.
-
-    Args:
-        _cls: The class to decorate
-
-    Returns:
-        The decorated class.
-    """
-    for _, value in _cls.__fields__.items():
-        value.required = False
-        value.allow_none = True
-
-    _cls.__config__.extra = Extra.ignore
-
-    return _cls