Use 50 GB Parquet+PyArrow dataset

coiled · Aug 15, 2024 · bbb4c71 · bbb4c71
1 parent 16dc57c
commit bbb4c71
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/AB_environments/config.yaml b/AB_environments/config.yaml
@@ -35,8 +35,8 @@ h2o_datasets:
   # - 0.5 GB (parquet)
   # - 5 GB (parquet)
   # - 50 GB (parquet)
-  - 5 GB (parquet+pyarrow)
-  # - 50 GB (parquet+pyarrow)
+  # - 5 GB (parquet+pyarrow)
+  - 50 GB (parquet+pyarrow)
   # - 500 GB (parquet+pyarrow)
 
 # AWS implements limiters to how many EC2 instances you can spawn in parallel on the

diff --git a/tests/benchmarks/test_h2o.py b/tests/benchmarks/test_h2o.py
@@ -31,7 +31,7 @@
         raise ValueError("Unknown h2o dataset(s): ", unknown_datasets)
 else:
     enabled_datasets = {
-        "5 GB (parquet)",
+        "50 GB (parquet+pyarrow)",
     }