Reformat examples

adap · Apr 3, 2024 · 8ad843f · 8ad843f
1 parent e429a70
commit 8ad843f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/examples/xgboost-comprehensive/dataset.py b/examples/xgboost-comprehensive/dataset.py
@@ -27,9 +27,7 @@ def instantiate_partitioner(partitioner_type: str, num_partitions: int):
 
 def train_test_split(partition: Dataset, test_fraction: float, seed: int):
     """Split the data into train and validation set given split rate."""
-    train_test = partition.train_test_split(
-        test_size=test_fraction, seed=seed, shuffle=False
-    )
+    train_test = partition.train_test_split(test_size=test_fraction, seed=seed)
     partition_train = train_test["train"]
     partition_test = train_test["test"]