From 8ad843fe986f7b87f4a01bdd6a29e76dd29f3048 Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Wed, 3 Apr 2024 22:55:39 +0200 Subject: [PATCH] Reformat examples --- examples/xgboost-comprehensive/dataset.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/examples/xgboost-comprehensive/dataset.py b/examples/xgboost-comprehensive/dataset.py index 21526b951cfb..94959925f833 100644 --- 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"]