wip

Author: fealho

sdmetrics/reports/single_table/_properties/column_pair_trends.py

@@ -454,19 +454,6 @@ def _get_correlation_matrix(self, column_name):
 
         return heatmap_df.astype(float).round(3)
 
-    def _get_heatmap_customdata(self, correlation_matrix, customdata):
-        if isinstance(customdata, pd.DataFrame):
-            customdata = customdata.reindex(
-                index=correlation_matrix.index,
-                columns=correlation_matrix.columns,
-            )
-            return customdata.to_numpy()
-
-        if isinstance(customdata, pd.Series):
-            return customdata.to_numpy()
-
-        return customdata
-
     def _get_heatmap(self, correlation_matrix, coloraxis, hovertemplate, customdata=None):
         """Get the heatmap for the given correlation matrix.
 
@@ -480,13 +467,12 @@ def _get_heatmap(self, correlation_matrix, coloraxis, hovertemplate, customdata=
             customdata (pandas.DataFrame or None):
                 The customdata to use. Defaults to None.
         """
-        customdata_array = self._get_heatmap_customdata(correlation_matrix, customdata)
         base_heatmap = go.Heatmap(
             x=correlation_matrix.columns,
             y=correlation_matrix.columns,
             z=correlation_matrix,
             coloraxis=coloraxis,
-            customdata=customdata_array,
+            customdata=customdata,
             hovertemplate=hovertemplate,
         )