digitalhub-runtime-python 0.16.0b1__tar.gz → 0.16.0b2__tar.gz

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalhub-runtime-python
-Version: 0.16.0b1
+Version: 0.16.0b2
 Summary: Python runtime for DHCore
 Project-URL: Homepage, https://github.com/scc-digitalhub/digitalhub-sdk-runtime-python
 Author-email: Fondazione Bruno Kessler <digitalhub@fbk.eu>, Matteo Martini <mmartini@fbk.eu>

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/__init__.py

@@ -97,4 +97,4 @@ except ImportError as e:
 
     logger = get_logger(__name__)
     logger.debug(f"Error importing runtime builders: {e}")
-    runtime_builders = tuple()
+    runtime_builders = ()

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/entities/function/guardrail/builder.py

@@ -34,6 +34,7 @@ class FunctionGuardrailBuilder(FunctionBuilder, RuntimeEntityBuilderGuardrail):
         project: str,
         name: str,
         uuid: str | None = None,
+        version: str | None = None,
         description: str | None = None,
         labels: list[str] | None = None,
         embedded: bool = False,
@@ -41,13 +42,14 @@ class FunctionGuardrailBuilder(FunctionBuilder, RuntimeEntityBuilderGuardrail):
     ) -> FunctionGuardrail:
         kwargs = source_check(**kwargs)
         obj = super().build(
-            kind,
-            project,
-            name,
-            uuid,
-            description,
-            labels,
-            embedded,
+            kind=kind,
+            project=project,
+            name=name,
+            uuid=uuid,
+            version=version,
+            description=description,
+            labels=labels,
+            embedded=embedded,
             **kwargs,
         )
         return source_post_check(obj)

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/entities/function/hydra/builder.py

@@ -32,6 +32,7 @@ class FunctionHydraBuilder(FunctionBuilder, RuntimeEntityBuilderHydra):
         project: str,
         name: str,
         uuid: str | None = None,
+        version: str | None = None,
         description: str | None = None,
         labels: list[str] | None = None,
         embedded: bool = False,
@@ -40,13 +41,14 @@ class FunctionHydraBuilder(FunctionBuilder, RuntimeEntityBuilderHydra):
         kwargs = source_check(**kwargs)
         kwargs = config_check(**kwargs)
         obj = super().build(
-            kind,
-            project,
-            name,
-            uuid,
-            description,
-            labels,
-            embedded,
+            kind=kind,
+            project=project,
+            name=name,
+            uuid=uuid,
+            version=version,
+            description=description,
+            labels=labels,
+            embedded=embedded,
             **kwargs,
         )
         return source_post_check(obj)

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/entities/function/openinference/builder.py

@@ -34,6 +34,7 @@ class FunctionOpeninferenceBuilder(FunctionBuilder, RuntimeEntityBuilderOpeninfe
         project: str,
         name: str,
         uuid: str | None = None,
+        version: str | None = None,
         description: str | None = None,
         labels: list[str] | None = None,
         embedded: bool = False,
@@ -41,13 +42,14 @@ class FunctionOpeninferenceBuilder(FunctionBuilder, RuntimeEntityBuilderOpeninfe
     ) -> FunctionOpeninference:
         kwargs = source_check(**kwargs)
         obj = super().build(
-            kind,
-            project,
-            name,
-            uuid,
-            description,
-            labels,
-            embedded,
+            kind=kind,
+            project=project,
+            name=name,
+            uuid=uuid,
+            version=version,
+            description=description,
+            labels=labels,
+            embedded=embedded,
             **kwargs,
         )
         return source_post_check(obj)

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/entities/function/python/builder.py

@@ -31,6 +31,7 @@ class FunctionPythonBuilder(FunctionBuilder, RuntimeEntityBuilderPython):
         project: str,
         name: str,
         uuid: str | None = None,
+        version: str | None = None,
         description: str | None = None,
         labels: list[str] | None = None,
         embedded: bool = False,
@@ -38,13 +39,14 @@ class FunctionPythonBuilder(FunctionBuilder, RuntimeEntityBuilderPython):
     ) -> FunctionPython:
         kwargs = source_check(**kwargs)
         obj = super().build(
-            kind,
-            project,
-            name,
-            uuid,
-            description,
-            labels,
-            embedded,
+            kind=kind,
+            project=project,
+            name=name,
+            uuid=uuid,
+            version=version,
+            description=description,
+            labels=labels,
+            embedded=embedded,
             **kwargs,
         )
         return source_post_check(obj)

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/entities/run/_base/entity.py

@@ -44,7 +44,7 @@ class RunBaseRun(Run):
         self.refresh()
         inputs = self.inputs(as_dict=True)
         if self.local_execution():
-            for _, v in inputs.items():
+            for v in inputs.values():
                 self.add_relationship(
                     relation=Relationship.CONSUMES.value,
                     dest=v.get("key"),

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/entities/run/_base/utils.py

@@ -4,7 +4,7 @@
 
 from __future__ import annotations
 
-from typing import Callable
+from collections.abc import Callable
 
 from digitalhub.entities._commons.enums import EntityTypes
 from digitalhub.entities.artifact.crud import get_artifact

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/runtimes/hydra_runtime.py

@@ -1,5 +1,9 @@
+# SPDX-FileCopyrightText: © 2025 DSLab - Fondazione Bruno Kessler
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import sys
-from typing import Callable
+from collections.abc import Callable
 
 import yaml
 from digitalhub.utils.generic_utils import decode_base64_string
@@ -144,7 +148,7 @@ class RuntimeHydraSubtask(RuntimePythonJob):
         args = super()._compose_args(func, spec, project)
         try:
             args["cfg_passthrough"] = OmegaConf.create(spec.get("parameters", {}).get("cfg_passthrough", {}))
-        except Exception as e:
+        except Exception as e:  # noqa: BLE001
             print(f"Failed to convert cfg to container. Exception: {e.__class__}. Error: {e.args}")
             args["cfg_passthrough"] = {}
         return args

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/runtimes/runtime.py

@@ -4,7 +4,8 @@
 
 from __future__ import annotations
 
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
 from digitalhub.context.api import get_context
 from digitalhub.runtimes._base import Runtime

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/utils/configuration.py

@@ -4,8 +4,8 @@
 
 from __future__ import annotations
 
+from collections.abc import Callable
 from pathlib import Path
-from typing import Callable
 
 from digitalhub.stores.data.api import get_store
 from digitalhub.utils.generic_utils import (

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/utils/inputs.py

@@ -6,7 +6,8 @@ from __future__ import annotations
 
 import inspect
 import typing
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
 from digitalhub.context.api import get_context
 from digitalhub.entities.project.crud import get_project

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/utils/outputs.py

@@ -9,22 +9,21 @@ import typing
 from dataclasses import dataclass
 from typing import Any
 
-from digitalhub import (
-    log_croissant,
-    log_generic_artifact,
-    log_generic_dataitem,
-    log_generic_model,
-    log_mlflow,
-    log_sklearn,
-    log_table,
-)
 from digitalhub.context.api import get_context
 from digitalhub.entities._commons.enums import EntityKinds, Relationship, State
 from digitalhub.entities.artifact._base.entity import Artifact
-from digitalhub.entities.artifact.crud import log_artifact
+from digitalhub.entities.artifact.artifact.crud import log_artifact
 from digitalhub.entities.dataitem._base.entity import Dataitem
-from digitalhub.entities.dataitem.crud import log_dataitem
+from digitalhub.entities.dataitem.croissant.crud import log_croissant
+from digitalhub.entities.dataitem.dataitem.crud import log_dataitem
+from digitalhub.entities.dataitem.table.crud import log_table
 from digitalhub.entities.model._base.entity import Model
+from digitalhub.entities.model.huggingface.crud import log_huggingface
+from digitalhub.entities.model.mlflow.crud import log_mlflow
+from digitalhub.entities.model.model.crud import log_model
+from digitalhub.entities.model.sklearn.crud import log_sklearn
+from digitalhub.entities.model.tvm_ir.crud import log_tvm_ir
+from digitalhub.entities.model.tvm_so.crud import log_tvm_so
 from digitalhub.stores.readers.data.api import get_supported_dataframes
 from digitalhub.utils.exceptions import EntityNotExistsError
 from digitalhub.utils.logger.logger import get_logger
@@ -36,13 +35,16 @@ logger = get_logger(__file__)
 
 
 mapped_logger = {
-    EntityKinds.DATAITEM_DATAITEM.value: log_generic_dataitem,
+    EntityKinds.DATAITEM_DATAITEM.value: log_dataitem,
     EntityKinds.DATAITEM_TABLE.value: log_table,
     EntityKinds.DATAITEM_CROISSANT.value: log_croissant,
     EntityKinds.MODEL_MLFLOW.value: log_mlflow,
     EntityKinds.MODEL_SKLEARN.value: log_sklearn,
-    EntityKinds.MODEL_MODEL.value: log_generic_model,
-    EntityKinds.ARTIFACT_ARTIFACT.value: log_generic_artifact,
+    EntityKinds.MODEL_MODEL.value: log_model,
+    EntityKinds.MODEL_HUGGINGFACE.value: log_huggingface,
+    EntityKinds.MODEL_TVM_IR.value: log_tvm_ir,
+    EntityKinds.MODEL_TVM_SO.value: log_tvm_so,
+    EntityKinds.ARTIFACT_ARTIFACT.value: log_artifact,
 }
 
 
@@ -97,7 +99,7 @@ def _parse_UDF_outputs(outputs: list[str] | list[dict[str, Any]], results_len: i
             continue
 
         if not isinstance(item, dict):
-            raise ValueError(f"Invalid output format: {item}")
+            raise TypeError(f"Invalid output format: {item}")
 
         name = item.get("name") or f"output_{idx}"
         kind = item.get("kind")
@@ -140,7 +142,7 @@ def _materialize_output(name: str, item: Any, project_name: str, run_key: str) -
 
     for df_class in get_supported_dataframes():
         if isinstance(item, df_class):
-            return _log_dataitem(name, project_name, item)
+            return _log_table(name, project_name, item)
 
     return _log_artifact(name, project_name, item)
 
@@ -232,7 +234,7 @@ def listify_results(results: Any) -> list:
     return results
 
 
-def _log_dataitem(name: str, project_name: str, data: Any) -> DataitemTable:
+def _log_table(name: str, project_name: str, data: Any) -> DataitemTable:
     """
     Log dataitem.
 
@@ -247,14 +249,13 @@ def _log_dataitem(name: str, project_name: str, data: Any) -> DataitemTable:
 
     Returns
     -------
-    str
-        Dataitem key.
+    DataitemTable
+        Dataitem table object.
     """
     try:
-        return log_dataitem(
+        return log_table(
             project=project_name,
             name=name,
-            kind=EntityKinds.DATAITEM_TABLE.value,
             data=data,
         )
     except Exception as e:
@@ -289,7 +290,6 @@ def _log_artifact(name: str, project_name: str, data: Any) -> Artifact:
         return log_artifact(
             project=project_name,
             name=name,
-            kind=EntityKinds.ARTIFACT_ARTIFACT.value,
             source=pickle_file,
         )
 

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/digitalhub_runtime_python/utils/utils.py

@@ -5,7 +5,7 @@
 from __future__ import annotations
 
 import functools
-from typing import Callable
+from collections.abc import Callable
 
 from digitalhub_runtime_python.utils.outputs import collect_outputs
 

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "digitalhub-runtime-python"
-version = "0.16.0b1"
+version = "0.16.0b2"
 description = "Python runtime for DHCore"
 readme = "README.md"
 authors = [
@@ -54,7 +54,7 @@ docstring-code-line-length = 120
 convention = "numpy"
 
 [tool.bumpver]
-current_version = "0.16.0b1"
+current_version = "0.16.0b2"
 version_pattern = "MAJOR.MINOR.PATCH[PYTAGNUM]"
 commit_message  = "Bump version {old_version} -> {new_version}"
 commit          = false

{digitalhub_runtime_python-0.16.0b1 → digitalhub_runtime_python-0.16.0b2}/test/instances/test_validate.py

@@ -4,7 +4,6 @@
 from __future__ import annotations
 
 import json
-from functools import lru_cache
 from pathlib import Path
 from typing import Any
 
@@ -24,7 +23,7 @@ class SchemaRegistry:
         self._schemas_path = schemas_path
         self._schema_paths = {path.stem: path for path in schemas_path.rglob("*.json")}
 
-    @lru_cache(maxsize=None)
+
     def get_schema(self, kind: str) -> dict[str, Any]:
         """Load and cache schema by kind."""
         if kind not in self._schema_paths: