Apply suggestions from code review

Co-authored-by: xiang song(charlie.song) <[email protected]>
awslabs · Oct 24, 2023 · aa1013d · aa1013d
1 parent b491ae3
commit aa1013d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"""