From 37ef667858e85fbd687ef34484e52e5f1fd44f26 Mon Sep 17 00:00:00 2001 From: "tom.mansion" Date: Mon, 27 Oct 2025 16:39:41 +0100 Subject: [PATCH 1/2] 1.1.5 Added columns metrics to interface --- debiai_data_provider/models/project.py | 4 +++- debiai_data_provider/version.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/debiai_data_provider/models/project.py b/debiai_data_provider/models/project.py index da04d0a..d95d842 100644 --- a/debiai_data_provider/models/project.py +++ b/debiai_data_provider/models/project.py @@ -64,6 +64,7 @@ def get_columns(self) -> Union[List[Column], None]: structure = structure.copy() + # Validate structure for key, value in structure.items(): if not isinstance(value, dict): raise ValueError( @@ -115,6 +116,7 @@ def get_columns(self) -> Union[List[Column], None]: # "type": "text", # "category": "context", # "group": "context", + # "metrics": {}, # }, # ... # } @@ -132,7 +134,7 @@ def get_columns(self) -> Union[List[Column], None]: "category": value["category"], "group": value.get("group", ""), }, - metrics={}, + metrics=value.get("metrics", {}), tags=[], type=value["type"], ) diff --git a/debiai_data_provider/version.py b/debiai_data_provider/version.py index 6886913..4981593 100644 --- a/debiai_data_provider/version.py +++ b/debiai_data_provider/version.py @@ -1 +1 @@ -VERSION = "1.1.4" +VERSION = "1.1.5" From 8a5197b6af7c2f488585fb06d4b3e3f466e4e082 Mon Sep 17 00:00:00 2001 From: "tom.mansion" Date: Tue, 28 Oct 2025 11:42:31 +0100 Subject: [PATCH 2/2] 1.1.6 Minor improvements --- debiai_data_provider/controller/routes.py | 3 +++ debiai_data_provider/models/project.py | 8 +++++--- debiai_data_provider/version.py | 2 +- 3 files changed, 9 insertions(+), 4 deletions(-) diff --git a/debiai_data_provider/controller/routes.py b/debiai_data_provider/controller/routes.py index 5745e31..346ce6a 100644 --- a/debiai_data_provider/controller/routes.py +++ b/debiai_data_provider/controller/routes.py @@ -104,6 +104,9 @@ async def get_data_id_list( ) def get_data( projectId: str = Path(..., min_length=1, example="Project 1"), + analysisId: Optional[str] = Query(None), + analysisStart: Optional[bool] = Query(None), + analysisEnd: Optional[bool] = Query(None), sampleIds: List[Union[str, int, float]] = Body(..., embed=True), data_provider: DataProvider = Depends(get_data_provider), ): diff --git a/debiai_data_provider/models/project.py b/debiai_data_provider/models/project.py index d95d842..f138fb1 100644 --- a/debiai_data_provider/models/project.py +++ b/debiai_data_provider/models/project.py @@ -33,7 +33,7 @@ def get_nb_samples(self) -> Union[int, None]: def get_samples_ids(self) -> List[str]: raise NotImplementedError - def get_data(self, samples_ids: List[str]) -> pd.DataFrame: + def get_data(self, samples_ids: List[Union[str, int, float]]) -> pd.DataFrame: raise NotImplementedError # Project models @@ -43,7 +43,9 @@ def get_models(self) -> List[ModelDetail]: def get_model_evaluated_data_id_list(self, model_id: str) -> List[str]: raise NotImplementedError - def get_model_results(self, model_id: str, sample_ids: List[str]) -> pd.DataFrame: + def get_model_results( + self, model_id: str, sample_ids: List[Union[str, int, float]] + ) -> pd.DataFrame: raise NotImplementedError @@ -323,7 +325,7 @@ def get_data_id_list( return samples_ids - def get_data_from_ids(self, samples_ids: List[str]) -> dict: + def get_data_from_ids(self, samples_ids: List[Union[str, int, float]]) -> dict: from debiai_data_provider.utils.parser import dataframe_to_debiai_data_array # Get the data from the project diff --git a/debiai_data_provider/version.py b/debiai_data_provider/version.py index 4981593..7a20704 100644 --- a/debiai_data_provider/version.py +++ b/debiai_data_provider/version.py @@ -1 +1 @@ -VERSION = "1.1.5" +VERSION = "1.1.6"