Remove correlation from DataQuality presets. (#1219)

evidentlyai · Jul 24, 2024 · fff2a0e · fff2a0e
1 parent ac03533
commit fff2a0e
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/src/evidently/metric_preset/data_quality.py b/src/evidently/metric_preset/data_quality.py
@@ -5,7 +5,6 @@
 
 from evidently.metric_preset.metric_preset import MetricPreset
 from evidently.metrics import ColumnSummaryMetric
-from evidently.metrics import DatasetCorrelationsMetric
 from evidently.metrics import DatasetSummaryMetric
 from evidently.metrics.base_metric import generate_column_metrics
 from evidently.metrics.data_integrity.dataset_missing_values_metric import DatasetMissingValuesMetric
@@ -36,5 +35,4 @@ def generate_metrics(self, data_definition: DataDefinition, additional_data: Opt
             DatasetSummaryMetric(),
             generate_column_metrics(ColumnSummaryMetric, columns=self.columns, skip_id_column=True),
             DatasetMissingValuesMetric(),
-            DatasetCorrelationsMetric(),
         ]
diff --git a/src/evidently/test_preset/data_quality.py b/src/evidently/test_preset/data_quality.py
@@ -6,7 +6,6 @@
 from evidently.test_preset.test_preset import TestPreset
 from evidently.tests import TestAllColumnsMostCommonValueShare
 from evidently.tests import TestAllColumnsShareOfMissingValues
-from evidently.tests import TestHighlyCorrelatedColumns
 from evidently.tests import TestNumberOfConstantColumns
 from evidently.tests import TestNumberOfDuplicatedColumns
 from evidently.tests import TestNumberOfDuplicatedRows
@@ -42,5 +41,4 @@ def generate_tests(self, data_definition: DataDefinition, additional_data: Optio
             TestNumberOfConstantColumns(),
             TestNumberOfDuplicatedColumns(),
             TestNumberOfDuplicatedRows(),
-            TestHighlyCorrelatedColumns(),
         ]
diff --git a/tests/test_preset/test_data_quality.py b/tests/test_preset/test_data_quality.py
@@ -29,4 +29,4 @@ def test_data_quality_preset():
 
     data_quality_suite.run(current_data=test_current_dataset, reference_data=test_reference_dataset)
     assert not data_quality_suite
-    assert len(data_quality_suite.as_dict()["tests"]) == 12
+    assert len(data_quality_suite.as_dict()["tests"]) == 11