rapidata 2.9.0__py3-none-any.whl → 2.10.0__py3-none-any.whl

rapidata/rapidata_client/order/rapidata_results.py

@@ -8,30 +8,40 @@ class RapidataResults(dict):
     A specialized dictionary class for handling Rapidata API results.
     Extends the built-in dict class with specialized methods.
     """
-    def to_pandas(self) -> pd.DataFrame:
+    def to_pandas(self, split_detailed: bool = False) -> pd.DataFrame:
         """
+        This method is currently under development.
+
         Converts the results to a pandas DataFrame.
         
-        For Compare results, creates standardized A/B columns for metrics like:
-        - aggregatedResults
-        - aggregatedResultsRatios
-        - summedUserScores
-        - summedUserScoresRatios
-        
+        For Compare results, creates standardized A/B columns for metrics.
         For regular results, flattens nested dictionaries into columns with underscore-separated names.
         
+        Args:
+            split_detailed: If True, splits each datapoint by its detailed results,
+                          creating a row for each response with global metrics copied.
+                          
         Returns:
             pd.DataFrame: A DataFrame containing the processed results
+            
+        Raises:
+            ValueError: If split_detailed is True but no detailed results are found
         """
         if "results" not in self or not self["results"]:
             return pd.DataFrame()
         
+        if self["info"].get("orderType") is None:
+            print("Warning: Results are old and Order type is not specified. Dataframe might be wrong.")
+        
+        # Check for detailed results if split_detailed is True
+        if split_detailed:
+            if not self._has_detailed_results():
+                raise ValueError("No detailed results found in the data")
+            return self._to_pandas_with_detailed_results()
+            
         if self["info"].get("orderType") == "Compare":
             return self._compare_to_pandas()
         
-        if self["info"].get("orderType") is None:
-            print("Warning: Results are old and Order type is not specified. Dataframe might be wrong.")
-
         # Get the structure from first item
         first_item = self["results"][0]
         columns = []
@@ -51,6 +61,68 @@ class RapidataResults(dict):
             
         return pd.DataFrame(data, columns=Index(columns))
     
+    def _has_detailed_results(self) -> bool:
+        """
+        Checks if the results contain detailed results.
+        
+        Returns:
+            bool: True if detailed results exist, False otherwise
+        """
+        if not self.get("results"):
+            return False
+        
+        first_result = self["results"][0]
+        return "detailedResults" in first_result and isinstance(first_result["detailedResults"], list)
+    
+    def _to_pandas_with_detailed_results(self) -> pd.DataFrame:
+        """
+        Converts results to a pandas DataFrame with detailed results split into separate rows.
+        
+        Returns:
+            pd.DataFrame: A DataFrame with one row per detailed result
+        """
+        rows = []
+        
+        for result in self["results"]:
+            # Get all non-detailed results fields
+            base_data = {k: v for k, v in result.items() if k != "detailedResults"}
+            
+            # Process each detailed result
+            for detailed_result in result["detailedResults"]:
+                row = base_data.copy()  # Copy base data for each detailed result
+                
+                # Add flattened detailed result data
+                flattened = self._flatten_dict(detailed_result)
+                for key, value in flattened.items():
+                    row[key] = value
+                
+                rows.append(row)
+        
+        return pd.DataFrame(rows)
+    
+    def _flatten_dict(self, d: dict[str, Any], parent_key: str = '') -> dict[str, Any]:
+        """
+        Flattens a nested dictionary into a single-level dictionary with underscore-separated keys.
+        
+        Args:
+            d: The dictionary to flatten
+            parent_key: The parent key for nested dictionaries
+            
+        Returns:
+            dict: A flattened dictionary
+        """
+        items: list[tuple[str, Any]] = []
+        
+        for key, value in d.items():
+            new_key = f"{parent_key}_{key}" if parent_key else key
+            
+            if isinstance(value, dict):
+                items.extend(self._flatten_dict(value, new_key).items())
+            else:
+                items.append((new_key, value))
+                
+        return dict(items)
+
     def _build_column_structure(
         self, 
         d: dict[str, Any], 
@@ -97,7 +169,7 @@ class RapidataResults(dict):
             d = d.get(key, {})
         return d.get(path[-1])
 
-    def _compare_to_pandas(self):
+    def _compare_to_pandas(self) -> pd.DataFrame:
         """
         Converts Compare results to a pandas DataFrame dynamically.
         """
@@ -132,7 +204,7 @@ class RapidataResults(dict):
             
         return pd.DataFrame(rows)
 
-    def to_json(self, path: str="./results.json"):
+    def to_json(self, path: str="./results.json") -> None:
         """
         Saves the results to a JSON file.
         

{rapidata-2.9.0.dist-info → rapidata-2.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rapidata
-Version: 2.9.0
+Version: 2.10.0
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 Author: Rapidata AG

{rapidata-2.9.0.dist-info → rapidata-2.10.0.dist-info}/RECORD

@@ -399,7 +399,7 @@ rapidata/rapidata_client/order/_rapidata_dataset.py,sha256=ipSLGEQm6Z5Gj0Z13SmE5
 rapidata/rapidata_client/order/_rapidata_order_builder.py,sha256=N6mqmgneJSsb_no_Ps9BG3EhDekLgKxFYpjDCN-VVeg,13095
 rapidata/rapidata_client/order/rapidata_order.py,sha256=XI9xeEtqZIwdAXLe9Pn4RO8iPNSoBToQ2WD62TRWzTE,9115
 rapidata/rapidata_client/order/rapidata_order_manager.py,sha256=4wW5xtmEtdewFGGwMWSMbdLZpHvygtLCfYNv4lcHGg8,27271
-rapidata/rapidata_client/order/rapidata_results.py,sha256=tCQgfv70FHJBcVsgAjqkYZB8ZvqrPGD_hlOmzArlLHg,5037
+rapidata/rapidata_client/order/rapidata_results.py,sha256=_Upy8IDEzkMJJ7QXz-MnZ3ONkhZd2uZ6KPW0SyJ7s40,7866
 rapidata/rapidata_client/rapidata_client.py,sha256=A9mnSX6wzVF9TxS1YH87hTi4jCn75dIuP3KZj5Y_vFg,1957
 rapidata/rapidata_client/referee/__init__.py,sha256=q0Hv9nmfEpyChejtyMLT8hWKL0vTTf_UgUXPYNJ-H6M,153
 rapidata/rapidata_client/referee/_base_referee.py,sha256=MdFOhdxt3sRnWXLDKLJZKFdVpjBGn9jypPnWWQ6msQA,496
@@ -447,7 +447,7 @@ rapidata/service/credential_manager.py,sha256=Of0BQs_V1T7rkrWX9groLX790nOknaARwn
 rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5AnfY14BW8,877
 rapidata/service/openapi_service.py,sha256=Z4NrAuilLlIWBdGOv6otz36tHS_vvU36w5jmvOUTmqo,3198
 rapidata/service/token_manager.py,sha256=JZ5YbR5Di8dO3H4kK11d0kzWlrXxjgCmeNkHA4AapCM,6425
-rapidata-2.9.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-2.9.0.dist-info/METADATA,sha256=7GbcyE9ZroILh90eiE2K7HuX_33Xoon2mZP8pRTjL_E,1146
-rapidata-2.9.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-rapidata-2.9.0.dist-info/RECORD,,
+rapidata-2.10.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-2.10.0.dist-info/METADATA,sha256=Dtdyhe-hoUqSfuARXuuqg2ilt1NVgCxB9X1wo1r_DFA,1147
+rapidata-2.10.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+rapidata-2.10.0.dist-info/RECORD,,