Skip to content

Commit 2ae3c39

Browse files
committed
Removed extra code.
1 parent f3c2757 commit 2ae3c39

File tree

1 file changed

+1
-13
lines changed

1 file changed

+1
-13
lines changed

sklbench/datasets/transformer.py

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -119,19 +119,7 @@ def split_and_transform_data(bench_case, data, data_description):
119119

120120
rank = MPI.COMM_WORLD.Get_rank()
121121
adjust_number = (math.sqrt(rank) * 0.003) + 1
122-
123-
if "y" in data:
124-
x_train, y_train = (
125-
x_train * adjust_number,
126-
y_train,
127-
)
128-
129-
x_test, y_test = (
130-
x_test * adjust_number,
131-
y_test,
132-
)
133-
else:
134-
x_test = x_test * adjust_number
122+
x_test = x_test * adjust_number
135123

136124
elif distributed_split == "rank_based":
137125
from mpi4py import MPI

0 commit comments

Comments
 (0)