Fixed minor issue with unit tests.
@@ -130,7 +130,7 @@ def get_fast_benchmark(
y = torch.from_numpy(dataset[1]).long()
train_X, test_X, train_y, test_y = train_test_split(
- X, y, train_size=0.6, shuffle=True, stratify=y, random_state=seed+1
+ X, y, train_size=0.6, shuffle=True, stratify=y, random_state=seed
)
train_dataset = TensorDataset(train_X, train_y)
@@ -141,7 +141,7 @@ def get_fast_benchmark(
5,
task_labels=use_task_labels,
shuffle=shuffle,
- seed=seed+2
+ seed=seed
)
return my_nc_benchmark
