From 3222aaff9b28c16b470f479420921a00aea749fa Mon Sep 17 00:00:00 2001 From: Anthony Mahanna <43019056+aMahanna@users.noreply.github.com> Date: Fri, 19 Jan 2024 09:46:38 -0500 Subject: [PATCH] proposal: increase default batch size (#16) * increase default batch size * nit: use `not` instead of `is False` * revert 3e03007 wrong branch... --- arango_datasets/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/arango_datasets/datasets.py b/arango_datasets/datasets.py index f6978cf..80e9a5c 100644 --- a/arango_datasets/datasets.py +++ b/arango_datasets/datasets.py @@ -27,7 +27,7 @@ class Datasets: def __init__( self, db: Database, - batch_size: int = 50, + batch_size: int = 1000, metadata_file: str = "https://arangodb-dataset-library-ml.s3.amazonaws.com/root_metadata.json", # noqa: E501 preserve_existing: bool = False, ):