[release][train] Fix tune_scalability_xgboost_sweep.aws (#43295)

Signed-off-by: Justin Yu <[email protected]>
ray-project · Feb 21, 2024 · 1fbe0e2 · 1fbe0e2
1 parent f880528
commit 1fbe0e2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/release/tune_tests/scalability_tests/workloads/test_xgboost_sweep.py b/release/tune_tests/scalability_tests/workloads/test_xgboost_sweep.py
@@ -17,6 +17,7 @@
 Acceptance criteria: Should run faster than 2600 seconds. Should run without
 errors.
 """
+
 from collections import Counter
 import json
 import os
@@ -81,6 +82,7 @@ def main():
  config=config,
  num_samples=num_samples,
  resources_per_trial=ray_params.get_tune_resources(),
+ storage_path="/mnt/cluster_storage",
  )
  time_taken = time.monotonic() - start_time