Fix parameter passing

tinkoff-ai · Jul 3, 2023 · e7d8a75 · e7d8a75
1 parent c3c59eb
commit e7d8a75
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/tests/test_transforms/test_inference/test_inverse_transform.py b/tests/test_transforms/test_inference/test_inverse_transform.py
@@ -1230,21 +1230,24 @@ def _test_inverse_transform_future_with_target(
             ),
             (
                 MRMRFeatureSelectionTransform(
-                    relevance_table=StatisticsRelevanceTable(), top_k=2, return_features=True
+                    relevance_table=StatisticsRelevanceTable(), top_k=2, fast_redundancy=False
                 ),
                 "ts_with_exog",
-                {"create": {"weekday", "monthday", "positive"}},
+                {},
             ),
             (
                 MRMRFeatureSelectionTransform(
-                    relevance_table=StatisticsRelevanceTable(), top_k=2, fast_redundancy=False
+                    relevance_table=StatisticsRelevanceTable(),
+                    top_k=2,
+                    return_features=True,
+                    fast_redundancy=True,
                 ),
                 "ts_with_exog",
-                {},
+                {"create": {"weekday", "monthday", "positive"}},
             ),
             (
                 MRMRFeatureSelectionTransform(
-                    relevance_table=StatisticsRelevanceTable(), top_k=2, return_features=False
+                    relevance_table=StatisticsRelevanceTable(), top_k=2, return_features=True, fast_redundancy=False
                 ),
                 "ts_with_exog",
                 {"create": {"weekday", "monthday", "positive"}},