From aa1013d052ffcd0d337e3b3db5d56f8c617cca87 Mon Sep 17 00:00:00 2001 From: Theodore Vasiloudis Date: Tue, 24 Oct 2023 15:25:54 -0700 Subject: [PATCH] Apply suggestions from code review Co-authored-by: xiang song(charlie.song) --- .../tests/test_dist_numerical_transformation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graphstorm-processing/tests/test_dist_numerical_transformation.py b/graphstorm-processing/tests/test_dist_numerical_transformation.py index 4cabb9710f..6fee80f6c0 100644 --- a/graphstorm-processing/tests/test_dist_numerical_transformation.py +++ b/graphstorm-processing/tests/test_dist_numerical_transformation.py @@ -178,7 +178,7 @@ def test_multi_numerical_transformation_with_special_delimiter( assert row["ratings"][1] == float(i + 1) -def test_multi_numerical_transformation_with_imputer( +def test_multi_numerical_transformation_with_mean_imputer( multi_num_df_with_missing: DataFrame, check_df_schema ): """Test multi-num with mean imputation"""