Rolled back the accidental changes to the ranked_based distributed_split.

Author: KateBlueSky

sklbench/datasets/transformer.py

@@ -146,8 +146,7 @@ def split_and_transform_data(bench_case, data, data_description):
 
             x_test = x_test[test_start:test_end] * adjust_number
 
-    elif distributed_split == "rank_based" or knn_split_train:
-
+    if distributed_split == "rank_based" or knn_split_train:
         comm = MPI.COMM_WORLD
         rank = comm.Get_rank()
         size = comm.Get_size()
@@ -159,7 +158,6 @@ def split_and_transform_data(bench_case, data, data_description):
         train_end = (1 + rank) * n_train // size
         test_start = rank * n_test // size
         test_end = (1 + rank) * n_test // size
-        x_train_rank = x_train[train_start:train_end]
 
         if "y" in data:
             x_train, y_train = (
@@ -171,7 +169,8 @@ def split_and_transform_data(bench_case, data, data_description):
         else:
             x_train = x_train[train_start:train_end]
             if distributed_split == "rank_based":
-                x_test = x_test[test_start:test_end] * adjust_number
+                x_test = x_test[test_start:test_end]
+
 
     device = get_bench_case_value(bench_case, "algorithm:device", None)
     common_data_format = get_bench_case_value(bench_case, "data:format", "pandas")