From 2911e30a40d96a27c87047b469987f379a9c857b Mon Sep 17 00:00:00 2001 From: alex-hse-repository Date: Mon, 10 Jul 2023 16:57:36 +0300 Subject: [PATCH] Fix naming --- tests/test_analysis/test_feature_selection/test_mrmr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_analysis/test_feature_selection/test_mrmr.py b/tests/test_analysis/test_feature_selection/test_mrmr.py index 44c52508c..c2caf7c71 100644 --- a/tests/test_analysis/test_feature_selection/test_mrmr.py +++ b/tests/test_analysis/test_feature_selection/test_mrmr.py @@ -33,7 +33,7 @@ def df_with_regressors() -> Dict[str, pd.DataFrame]: regressor = df_regressors_useless[df_regressors_useless["segment"] == segment]["target"].values df_exog[f"regressor_useless_{i}"] = regressor - # useless categorical regressor + # useless categorical regressors num_cat_useless = 3 for i in range(num_cat_useless): df_exog[f"categorical_regressor_useless_{i}"] = i @@ -184,7 +184,7 @@ def test_fast_redundancy_deprecation_warning(df_with_regressors): @pytest.mark.parametrize("fast_redundancy", [True, False]) -def test_mrmr_with_categorical_regressor(df_with_regressors, fast_redundancy): +def test_mrmr_with_castable_categorical_regressor(df_with_regressors, fast_redundancy): df, regressors = df_with_regressors["df"], df_with_regressors["regressors"] relevance_table = ModelRelevanceTable()(df=df, df_exog=regressors, model=RandomForestRegressor()) mrmr(relevance_table=relevance_table, regressors=regressors, top_k=len(regressors), fast_redundancy=fast_redundancy)