proposal: increase default batch size (#16)

* increase default batch size * nit: use `not` instead of `is False` * revert 3e03007 wrong branch...
arangoml · Jan 19, 2024 · 3222aaf · 3222aaf
1 parent 2b8febe
commit 3222aaf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
     ):