digitalhub 0.8.0__py3-none-any.whl → 0.8.0b1__py3-none-any.whl

digitalhub/entities/model/_base/spec.py

@@ -1,49 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities._base.material.spec import MaterialSpec, MaterialValidator
-
-
-class ModelSpec(MaterialSpec):
-    """
-    ModelSpec specifications.
-    """
-
-    def __init__(
-        self,
-        path: str,
-        framework: str | None = None,
-        algorithm: str | None = None,
-        base_model: str | None = None,
-        parameters: dict | None = None,
-        metrics: dict | None = None,
-    ) -> None:
-        self.path = path
-        self.framework = framework
-        self.algorithm = algorithm
-        self.base_model = base_model
-        self.parameters = parameters
-        self.metrics = metrics
-
-
-class ModelValidator(MaterialValidator):
-    """
-    ModelValidator validator.
-    """
-
-    path: str
-    """Path to the model."""
-
-    framework: str = None
-    """Model framework (e.g. 'pytorch')."""
-
-    algorithm: str = None
-    """Model algorithm (e.g. 'resnet')."""
-
-    base_model: str = None
-    """Base model."""
-
-    parameters: dict = None
-    """Model validator."""
-
-    metrics: dict = None
-    """Model metrics."""

digitalhub/entities/model/_base/status.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities._base.material.status import MaterialStatus
-
-
-class ModelStatus(MaterialStatus):
-    """
-    ModelStatus status.
-    """

digitalhub/entities/model/huggingface/builder.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.builder import ModelBuilder
-from digitalhub.entities.model.huggingface.entity import ModelHuggingface
-from digitalhub.entities.model.huggingface.spec import ModelSpecHuggingface, ModelValidatorHuggingface
-from digitalhub.entities.model.huggingface.status import ModelStatusHuggingface
-
-
-class ModelHuggingfaceBuilder(ModelBuilder):
-    """
-    ModelHuggingface builder.
-    """
-
-    ENTITY_CLASS = ModelHuggingface
-    ENTITY_SPEC_CLASS = ModelSpecHuggingface
-    ENTITY_SPEC_VALIDATOR = ModelValidatorHuggingface
-    ENTITY_STATUS_CLASS = ModelStatusHuggingface
-    ENTITY_KIND = "huggingface"

digitalhub/entities/model/huggingface/entity.py

@@ -1,32 +0,0 @@
-from __future__ import annotations
-
-import typing
-
-from digitalhub.entities.model._base.entity import Model
-
-if typing.TYPE_CHECKING:
-    from digitalhub.entities._base.entity.metadata import Metadata
-    from digitalhub.entities.model.huggingface.spec import ModelSpecHuggingface
-    from digitalhub.entities.model.huggingface.status import ModelStatusHuggingface
-
-
-class ModelHuggingface(Model):
-    """
-    ModelHuggingface class.
-    """
-
-    def __init__(
-        self,
-        project: str,
-        name: str,
-        uuid: str,
-        kind: str,
-        metadata: Metadata,
-        spec: ModelSpecHuggingface,
-        status: ModelStatusHuggingface,
-        user: str | None = None,
-    ) -> None:
-        super().__init__(project, name, uuid, kind, metadata, spec, status, user)
-
-        self.spec: ModelSpecHuggingface
-        self.status: ModelStatusHuggingface

digitalhub/entities/model/huggingface/spec.py

@@ -1,36 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.spec import ModelSpec, ModelValidator
-
-
-class ModelSpecHuggingface(ModelSpec):
-    """
-    ModelSpecHuggingface specifications.
-    """
-
-    def __init__(
-        self,
-        path: str,
-        framework: str | None = None,
-        algorithm: str | None = None,
-        base_model: str | None = None,
-        parameters: dict | None = None,
-        metrics: dict | None = None,
-        model_id: str | None = None,
-        model_revision: str = None,
-    ) -> None:
-        super().__init__(path, framework, algorithm, base_model, parameters, metrics)
-        self.model_id = model_id
-        self.model_revision = model_revision
-
-
-class ModelValidatorHuggingface(ModelValidator):
-    """
-    ModelValidatorHuggingface validator.
-    """
-
-    model_id: str = None
-    """Huggingface model id. If not specified, the model is loaded from the model path."""
-
-    model_revision: str = None
-    """Huggingface model revision."""

digitalhub/entities/model/huggingface/status.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.status import ModelStatus
-
-
-class ModelStatusHuggingface(ModelStatus):
-    """
-    ModelStatusHuggingface status.
-    """

digitalhub/entities/model/mlflow/builder.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.builder import ModelBuilder
-from digitalhub.entities.model.mlflow.entity import ModelMlflow
-from digitalhub.entities.model.mlflow.spec import ModelSpecMlflow, ModelValidatorMlflow
-from digitalhub.entities.model.mlflow.status import ModelStatusMlflow
-
-
-class ModelModelBuilder(ModelBuilder):
-    """
-    ModelMlflow builder.
-    """
-
-    ENTITY_CLASS = ModelMlflow
-    ENTITY_SPEC_CLASS = ModelSpecMlflow
-    ENTITY_SPEC_VALIDATOR = ModelValidatorMlflow
-    ENTITY_STATUS_CLASS = ModelStatusMlflow
-    ENTITY_KIND = "mlflow"

digitalhub/entities/model/mlflow/entity.py

@@ -1,32 +0,0 @@
-from __future__ import annotations
-
-import typing
-
-from digitalhub.entities.model._base.entity import Model
-
-if typing.TYPE_CHECKING:
-    from digitalhub.entities._base.entity.metadata import Metadata
-    from digitalhub.entities.model.mlflow.spec import ModelSpecMlflow
-    from digitalhub.entities.model.mlflow.status import ModelStatusMlflow
-
-
-class ModelMlflow(Model):
-    """
-    ModelMlflow class.
-    """
-
-    def __init__(
-        self,
-        project: str,
-        name: str,
-        uuid: str,
-        kind: str,
-        metadata: Metadata,
-        spec: ModelSpecMlflow,
-        status: ModelStatusMlflow,
-        user: str | None = None,
-    ) -> None:
-        super().__init__(project, name, uuid, kind, metadata, spec, status, user)
-
-        self.spec: ModelSpecMlflow
-        self.status: ModelStatusMlflow

digitalhub/entities/model/mlflow/spec.py

@@ -1,44 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.spec import ModelSpec, ModelValidator
-from digitalhub.entities.model.mlflow.models import Dataset, Signature
-
-
-class ModelSpecMlflow(ModelSpec):
-    """
-    ModelSpecMlflow specifications.
-    """
-
-    def __init__(
-        self,
-        path: str,
-        framework: str | None = None,
-        algorithm: str | None = None,
-        base_model: str | None = None,
-        parameters: dict | None = None,
-        metrics: dict | None = None,
-        flavor: str | None = None,
-        model_config: dict | None = None,
-        input_datasets: list[Dataset] | None = None,
-        signature: Signature = None,
-    ) -> None:
-        super().__init__(path, framework, algorithm, base_model, parameters, metrics)
-        self.flavor = flavor
-        self.model_config = model_config
-        self.input_datasets = input_datasets
-        self.signature = signature
-
-
-class ModelValidatorMlflow(ModelValidator):
-    """
-    ModelValidatorMlflow validator.
-    """
-
-    flavor: str = None
-    """Mlflow model flavor."""
-    model_config: dict = None
-    """Mlflow model config."""
-    input_datasets: list[Dataset] = None
-    """Mlflow input datasets."""
-    signature: Signature = None
-    """Mlflow model signature."""

digitalhub/entities/model/mlflow/status.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.status import ModelStatus
-
-
-class ModelStatusMlflow(ModelStatus):
-    """
-    ModelStatusMlflow status.
-    """

digitalhub/entities/model/model/builder.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.builder import ModelBuilder
-from digitalhub.entities.model.model.entity import ModelModel
-from digitalhub.entities.model.model.spec import ModelSpecModel, ModelValidatorModel
-from digitalhub.entities.model.model.status import ModelStatusModel
-
-
-class ModelMlflowBuilder(ModelBuilder):
-    """
-    ModelModel builder.
-    """
-
-    ENTITY_CLASS = ModelModel
-    ENTITY_SPEC_CLASS = ModelSpecModel
-    ENTITY_SPEC_VALIDATOR = ModelValidatorModel
-    ENTITY_STATUS_CLASS = ModelStatusModel
-    ENTITY_KIND = "model"

digitalhub/entities/model/model/entity.py

@@ -1,32 +0,0 @@
-from __future__ import annotations
-
-import typing
-
-from digitalhub.entities.model._base.entity import Model
-
-if typing.TYPE_CHECKING:
-    from digitalhub.entities._base.entity.metadata import Metadata
-    from digitalhub.entities.model.model.spec import ModelSpecModel
-    from digitalhub.entities.model.model.status import ModelStatusModel
-
-
-class ModelModel(Model):
-    """
-    ModelModel class.
-    """
-
-    def __init__(
-        self,
-        project: str,
-        name: str,
-        uuid: str,
-        kind: str,
-        metadata: Metadata,
-        spec: ModelSpecModel,
-        status: ModelStatusModel,
-        user: str | None = None,
-    ) -> None:
-        super().__init__(project, name, uuid, kind, metadata, spec, status, user)
-
-        self.spec: ModelSpecModel
-        self.status: ModelStatusModel

digitalhub/entities/model/model/spec.py

@@ -1,15 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.spec import ModelSpec, ModelValidator
-
-
-class ModelSpecModel(ModelSpec):
-    """
-    ModelSpecModel specifications.
-    """
-
-
-class ModelValidatorModel(ModelValidator):
-    """
-    ModelValidatorModel validator.
-    """

digitalhub/entities/model/model/status.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.status import ModelStatus
-
-
-class ModelStatusModel(ModelStatus):
-    """
-    ModelStatusModel status.
-    """

digitalhub/entities/model/sklearn/builder.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.builder import ModelBuilder
-from digitalhub.entities.model.sklearn.entity import ModelSklearn
-from digitalhub.entities.model.sklearn.spec import ModelSpecSklearn, ModelValidatorSklearn
-from digitalhub.entities.model.sklearn.status import ModelStatusSklearn
-
-
-class ModelSklearnBuilder(ModelBuilder):
-    """
-    ModelSklearn builder.
-    """
-
-    ENTITY_CLASS = ModelSklearn
-    ENTITY_SPEC_CLASS = ModelSpecSklearn
-    ENTITY_SPEC_VALIDATOR = ModelValidatorSklearn
-    ENTITY_STATUS_CLASS = ModelStatusSklearn
-    ENTITY_KIND = "sklearn"

digitalhub/entities/model/sklearn/entity.py

@@ -1,32 +0,0 @@
-from __future__ import annotations
-
-import typing
-
-from digitalhub.entities.model._base.entity import Model
-
-if typing.TYPE_CHECKING:
-    from digitalhub.entities._base.entity.metadata import Metadata
-    from digitalhub.entities.model.sklearn.spec import ModelSpecSklearn
-    from digitalhub.entities.model.sklearn.status import ModelStatusSklearn
-
-
-class ModelSklearn(Model):
-    """
-    ModelSklearn class.
-    """
-
-    def __init__(
-        self,
-        project: str,
-        name: str,
-        uuid: str,
-        kind: str,
-        metadata: Metadata,
-        spec: ModelSpecSklearn,
-        status: ModelStatusSklearn,
-        user: str | None = None,
-    ) -> None:
-        super().__init__(project, name, uuid, kind, metadata, spec, status, user)
-
-        self.spec: ModelSpecSklearn
-        self.status: ModelStatusSklearn

digitalhub/entities/model/sklearn/spec.py

@@ -1,15 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.spec import ModelSpec, ModelValidator
-
-
-class ModelSpecSklearn(ModelSpec):
-    """
-    ModelSpecSklearn specifications.
-    """
-
-
-class ModelValidatorSklearn(ModelValidator):
-    """
-    ModelValidatorSklearn validator.
-    """

digitalhub/entities/model/sklearn/status.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities.model._base.status import ModelStatus
-
-
-class ModelStatusSklearn(ModelStatus):
-    """
-    ModelStatusSklearn status.
-    """

digitalhub/entities/project/_base/builder.py

@@ -1,128 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities._base.entity.builder import EntityBuilder
-from digitalhub.entities.project._base.entity import Project
-from digitalhub.entities.project._base.spec import ProjectSpec, ProjectValidator
-from digitalhub.entities.project._base.status import ProjectStatus
-from digitalhub.entities.utils.entity_types import EntityTypes
-
-
-class ProjectProjectBuilder(EntityBuilder):
-    """
-    ProjectProject builder.
-    """
-
-    ENTITY_TYPE = EntityTypes.PROJECT.value
-    ENTITY_CLASS = Project
-    ENTITY_SPEC_CLASS = ProjectSpec
-    ENTITY_SPEC_VALIDATOR = ProjectValidator
-    ENTITY_STATUS_CLASS = ProjectStatus
-    ENTITY_KIND = "project"
-
-    def build(
-        self,
-        name: str,
-        kind: str,
-        description: str | None = None,
-        labels: list[str] | None = None,
-        local: bool = False,
-        context: str | None = None,
-        **kwargs,
-    ) -> Project:
-        """
-        Create a new object.
-
-        Parameters
-        ----------
-        name : str
-            Object name.
-        kind : str
-            Kind the object.
-        description : str
-            Description of the object (human readable).
-        labels : list[str]
-            List of labels.
-        local : bool
-            If True, use local backend, if False use DHCore backend. Default to False.
-        context : str
-            The context local folder of the project.
-        **kwargs : dict
-            Spec keyword arguments.
-
-        Returns
-        -------
-        Project
-            Object instance.
-        """
-        name = self.build_name(name)
-        metadata = self.build_metadata(
-            project=name,
-            name=name,
-            description=description,
-            labels=labels,
-        )
-        spec = self.build_spec(
-            context=context,
-            **kwargs,
-        )
-        status = self.build_status()
-        return self.build_entity(
-            name=name,
-            kind=kind,
-            metadata=metadata,
-            spec=spec,
-            status=status,
-            local=local,
-        )
-
-    def from_dict(self, obj: dict, validate: bool = True) -> Project:
-        """
-        Create a new object from dictionary.
-
-        Parameters
-        ----------
-        obj : dict
-            Dictionary to create object from.
-        validate : bool
-            Flag to indicate if arguments must be validated.
-
-        Returns
-        -------
-        Project
-            Object instance.
-        """
-        parsed_dict = self._parse_dict(obj, validate=validate)
-        return self.build_entity(**parsed_dict)
-
-    def _parse_dict(self, obj: dict, validate: bool = True) -> dict:
-        """
-        Get dictionary and parse it to a valid entity dictionary.
-
-        Parameters
-        ----------
-        entity : str
-            Entity type.
-        obj : dict
-            Dictionary to parse.
-
-        Returns
-        -------
-        dict
-            A dictionary containing the attributes of the entity instance.
-        """
-        name = self.build_name(obj.get("name"))
-        kind = obj.get("kind")
-        local = obj.get("local", False)
-        metadata = self.build_metadata(**obj.get("metadata", {}))
-        spec = self.build_spec(validate=validate, **obj.get("spec", {}))
-        status = self.build_status(**obj.get("status", {}))
-        user = obj.get("user")
-        return {
-            "name": name,
-            "kind": kind,
-            "metadata": metadata,
-            "spec": spec,
-            "status": status,
-            "user": user,
-            "local": local,
-        }

digitalhub/entities/project/_base/status.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from digitalhub.entities._base.entity.status import Status
-
-
-class ProjectStatus(Status):
-    """
-    ProjectStatus status.
-    """

digitalhub/entities/run/_base/builder.py

@@ -1,94 +0,0 @@
-from __future__ import annotations
-
-import typing
-
-from digitalhub.entities._base.runtime_entity.builder import EntityError, RuntimeEntityBuilder
-from digitalhub.entities._base.unversioned.builder import UnversionedBuilder
-from digitalhub.entities.utils.entity_types import EntityTypes
-
-if typing.TYPE_CHECKING:
-    from digitalhub.entities.run._base.entity import Run
-
-
-class RunBuilder(UnversionedBuilder, RuntimeEntityBuilder):
-    """
-    Run builder.
-    """
-
-    ENTITY_TYPE = EntityTypes.RUN.value
-
-    def build(
-        self,
-        project: str,
-        kind: str,
-        uuid: str | None = None,
-        labels: list[str] | None = None,
-        task: str | None = None,
-        local_execution: bool = False,
-        **kwargs,
-    ) -> Run:
-        """
-        Create a new object.
-
-        Parameters
-        ----------
-        project : str
-            Project name.
-        kind : str
-            Kind the object.
-        uuid : str
-            ID of the object.
-        labels : list[str]
-            List of labels.
-        task : str
-            Name of the task associated with the run.
-        local_execution : bool
-            Flag to determine if object has local execution.
-        **kwargs : dict
-            Spec keyword arguments.
-
-        Returns
-        -------
-        Run
-            Object instance.
-        """
-        if task is None:
-            raise EntityError("Missing task in run spec")
-        self._check_kind_validity(task)
-        uuid = self.build_uuid(uuid)
-        metadata = self.build_metadata(
-            project=project,
-            name=uuid,
-            labels=labels,
-        )
-        spec = self.build_spec(
-            task=task,
-            local_execution=local_execution,
-            **kwargs,
-        )
-        status = self.build_status()
-        return self.build_entity(
-            project=project,
-            uuid=uuid,
-            kind=kind,
-            metadata=metadata,
-            spec=spec,
-            status=status,
-        )
-
-    def _check_kind_validity(self, task: str) -> None:
-        """
-        Check kind validity.
-
-        Parameters
-        ----------
-        task : str
-            Task string.
-
-        Returns
-        -------
-        None
-        """
-        task_kind = task.split("://")[0]
-        if task_kind not in self.get_all_kinds():
-            raise EntityError(f"Invalid run '{self.ENTITY_KIND}' for task kind '{task_kind}'")