digitalhub 0.14.0b5__py3-none-any.whl → 0.14.9__py3-none-any.whl

digitalhub/entities/dataitem/utils.py

@@ -41,17 +41,17 @@ def eval_source(
 
     Parameters
     ----------
-    source : SourcesOrListOfSources, optional
+    source : SourcesOrListOfSources
         The source specification(s) for the dataitem. Can be file paths,
         URLs, or other source identifiers.
-    data : Any, optional
+    data : Any
         The data object to process (e.g., DataFrame). Alternative to source.
         Exactly one of source or data must be provided.
-    kind : str, optional
+    kind : str
         The kind of dataitem being created (e.g., 'table').
-    name : str, optional
+    name : str
         The name of the dataitem, used for generating file paths.
-    project : str, optional
+    project : str
         The project name, used for context and path generation.
 
     Returns
@@ -105,13 +105,13 @@ def eval_data(
         how data should be processed.
     source : SourcesOrListOfSources
         The source specification(s) to load data from.
-    data : Any, optional
+    data : Any
         Pre-loaded data object. If provided, may be returned directly
         depending on the dataitem kind.
-    file_format : str, optional
+    file_format : str
         The file format specification for reading the source
         (e.g., 'parquet', 'csv').
-    engine : str, optional
+    engine : str
         The engine to use for reading the data (e.g., 'pandas', 'polars').
 
     Returns
@@ -159,10 +159,10 @@ def process_kwargs(
         The kind of dataitem being created (e.g., 'table').
     source : SourcesOrListOfSources
         The source specification(s) for the dataitem content.
-    data : Any, optional
+    data : Any
         The data object for schema extraction and processing.
         Used as an alternative to source for table dataitems.
-    path : str, optional
+    path : str
         The destination path for the dataitem entity.
         If None, a path will be automatically generated.
     **kwargs : dict
@@ -201,10 +201,6 @@ def clean_tmp_path(pth: SourcesOrListOfSources) -> None:
     pth : SourcesOrListOfSources
         The path or list of paths to clean up. Can be file paths
         or directory paths that need to be removed.
-
-    Returns
-    -------
-    None
     """
     if isinstance(pth, list):
         for p in pth:

digitalhub/entities/function/_base/entity.py

@@ -10,7 +10,6 @@ from concurrent.futures import ThreadPoolExecutor
 from digitalhub.entities._base.executable.entity import ExecutableEntity
 from digitalhub.entities._commons.enums import EntityTypes, Relationship
 from digitalhub.factory.entity import entity_factory
-from digitalhub.utils.exceptions import BackendError
 
 if typing.TYPE_CHECKING:
     from digitalhub.entities._base.entity.metadata import Metadata
@@ -91,8 +90,6 @@ class Function(ExecutableEntity):
 
         # If execution is done by DHCore backend, return the object
         if not local_execution:
-            if self._context().local:
-                raise BackendError("Cannot run remote function with local backend.")
             if wait:
                 return run.wait(log_info=log_info)
             return run

digitalhub/entities/function/crud.py

@@ -156,7 +156,7 @@ def list_functions(
     state: str | None = None,
     created: str | None = None,
     updated: str | None = None,
-    version: str | None = None,
+    versions: str | None = None,
 ) -> list[Function]:
     """
     List all latest version objects from backend.
@@ -179,7 +179,7 @@ def list_functions(
         Creation date filter.
     updated : str
         Update date filter.
-    version : str
+    versions : str
         Object version, default is latest.
 
     Returns
@@ -201,7 +201,7 @@ def list_functions(
         state=state,
         created=created,
         updated=updated,
-        version=version,
+        versions=versions,
     )
 
 

digitalhub/entities/model/crud.py

@@ -211,7 +211,7 @@ def list_models(
     state: str | None = None,
     created: str | None = None,
     updated: str | None = None,
-    version: str | None = None,
+    versions: str | None = None,
 ) -> list[Model]:
     """
     List all latest version objects from backend.
@@ -234,7 +234,7 @@ def list_models(
         Creation date filter.
     updated : str
         Update date filter.
-    version : str
+    versions : str
         Object version, default is latest.
 
     Returns
@@ -256,7 +256,7 @@ def list_models(
         state=state,
         created=created,
         updated=updated,
-        version=version,
+        versions=versions,
     )
 
 

digitalhub/entities/model/mlflow/utils.py

@@ -53,10 +53,13 @@ def from_mlflow_run(run_id: str) -> dict:
     # Extract spec
     data = run.data
     parameters = data.params
-    source_path = urlparse(run.info.artifact_uri).path
+    source_path = urlparse(run.info.artifact_uri).path + "/model"
     model_uri = f"runs:/{run_id}/model"
     model = mlflow.pyfunc.load_model(model_uri=model_uri)
-    model_config = model.model_config
+    try:
+        model_config = model.model_config
+    except Exception:
+        model_config = {}
     flavor = None
     for f in model.metadata.flavors:
         if f != "python_function":
@@ -64,27 +67,33 @@ def from_mlflow_run(run_id: str) -> dict:
             break
 
     # Extract signature
-    mlflow_signature = model.metadata.signature
-    signature = Signature(
-        inputs=mlflow_signature.inputs.to_json() if mlflow_signature.inputs else None,
-        outputs=mlflow_signature.outputs.to_json() if mlflow_signature.outputs else None,
-        params=mlflow_signature.params.to_json() if mlflow_signature.params else None,
-    ).to_dict()
+    try:
+        mlflow_signature = model.metadata.signature
+        signature = Signature(
+            inputs=mlflow_signature.inputs.to_json() if mlflow_signature.inputs else None,
+            outputs=mlflow_signature.outputs.to_json() if mlflow_signature.outputs else None,
+            params=mlflow_signature.params.to_json() if mlflow_signature.params else None,
+        ).to_dict()
+    except Exception:
+        signature = None
 
     # Extract datasets
     datasets = []
-    if run.inputs and run.inputs.dataset_inputs:
-        datasets = [
-            Dataset(
-                name=d.dataset.name,
-                digest=d.dataset.digest,
-                profile=d.dataset.profile,
-                dataset_schema=d.dataset.schema,
-                source=d.dataset.source,
-                source_type=d.dataset.source_type,
-            ).to_dict()
-            for d in run.inputs.dataset_inputs
-        ]
+    try:
+        if run.inputs and run.inputs.dataset_inputs:
+            datasets = [
+                Dataset(
+                    name=d.dataset.name,
+                    digest=d.dataset.digest,
+                    profile=d.dataset.profile,
+                    dataset_schema=d.dataset.schema,
+                    source=d.dataset.source,
+                    source_type=d.dataset.source_type,
+                ).to_dict()
+                for d in run.inputs.dataset_inputs
+            ]
+    except Exception:
+        datasets = []
 
     # Create model params
     model_params = {}

digitalhub/entities/model/utils.py

@@ -23,7 +23,7 @@ def eval_source(
 
     Parameters
     ----------
-    source : str, list[str], or None, optional
+    source : str, list[str], or None
         The source specification(s) to evaluate. Can be a single
         source string, a list of source strings, or None.
 
@@ -59,7 +59,7 @@ def process_kwargs(
     source : str or list[str]
         The source specification(s) for the model content.
         Can be a single source or multiple sources.
-    path : str, optional
+    path : str
         The destination path for the model entity.
         If None, a path will be automatically generated.
     **kwargs : dict

digitalhub/entities/project/_base/builder.py

@@ -29,7 +29,6 @@ class ProjectProjectBuilder(EntityBuilder):
         kind: str,
         description: str | None = None,
         labels: list[str] | None = None,
-        local: bool = False,
         config: dict | None = None,
         context: str | None = None,
         **kwargs,
@@ -47,8 +46,6 @@ class ProjectProjectBuilder(EntityBuilder):
             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.
         config : dict
             DHCore environment configuration.
         context : str
@@ -80,7 +77,6 @@ class ProjectProjectBuilder(EntityBuilder):
             metadata=metadata,
             spec=spec,
             status=status,
-            local=local,
         )
 
     def from_dict(self, obj: dict, validate: bool = True) -> Project:
@@ -120,7 +116,6 @@ class ProjectProjectBuilder(EntityBuilder):
         """
         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", {}))
@@ -132,5 +127,4 @@ class ProjectProjectBuilder(EntityBuilder):
             "spec": spec,
             "status": status,
             "user": user,
-            "local": local,
         }