utils: move get_raw() method into DSSInternalDict

Author: jfyuen

dataikuapi/dss/ml.py

@@ -777,12 +777,6 @@ def __init__(self, feature_name, computed_as_type, data, prediction_type):
         elif computed_as_type == "NUMERIC":
             self.definition = DSSSubpopulationNumericModalityDefinition(feature_name, data)
 
-    def get_raw(self):
-        """
-        Gets the raw dictionary of the subpopulation analysis modality
-        """
-        return self._internal_dict
-    
     def get_definition(self):
         """
         Gets the definition of the subpopulation analysis modality
@@ -950,12 +944,6 @@ def get_modality_data(self, definition=None):
                 return m
         raise ValueError("Modality not found: %s" % definition)
 
-    def get_raw(self):
-        """
-        Gets the raw dictionary of the subpopulation analysis
-        """
-        return self._internal_dict
-
 
 class DSSSubpopulationAnalyses(DSSInternalDict):
     """
@@ -971,12 +959,6 @@ def __init__(self, data, prediction_type):
         for analysis in data.get("subpopulationAnalyses", []):
             self.analyses.append(DSSSubpopulationAnalysis(analysis, prediction_type))
 
-    def get_raw(self):
-        """
-        Gets the raw dictionary of subpopulation analyses
-        """
-        return self._internal_dict
-    
     def get_global(self):
         """
         Retrieves information and performance on the full dataset used to compute the subpopulation analyses
@@ -1019,12 +1001,6 @@ def get_computation_params(self):
             "onSample":  self.get("onSample")
         }
 
-    def get_raw(self):
-        """
-        Gets the raw dictionary of the partial dependence
-        """
-        return self._internal_dict
-
 
 class DSSPartialDependencies(DSSInternalDict):
     """
@@ -1039,12 +1015,6 @@ def __init__(self, data):
         for pd in data.get("partialDependencies", []):
             self.partial_dependencies.append(DSSPartialDependence(pd))
 
-    def get_raw(self):
-        """
-        Gets the raw dictionary of partial dependencies
-        """
-        return self._internal_dict
-
     def list_features(self):
         """
         Lists all features on which partial dependencies have been computed

dataikuapi/utils.py

@@ -109,5 +109,13 @@ def __init__(self, orig_dict=None):
     def get(self, name, default=None):
         return self._internal_dict.get(name, default=default)
 
+    def get_raw(self):
+        """
+        Gets the raw dictionary of the actual data
+
+        :rtype: dict
+        """
+        return self._internal_dict
+
     def __repr__(self):
         return self.__class__.__name__ + "(" + self._internal_dict.__repr__() + ")"