From fdc56b33863c4cbc988b30bceebbec7b6909d8a3 Mon Sep 17 00:00:00 2001
From: Favyen Bastani <favyenb@allenai.org>
Date: Thu, 6 Feb 2025 11:04:56 -0800
Subject: [PATCH] add CLUSTERS constant

---
 rslp/sentinel2_vessels/job_launcher.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/rslp/sentinel2_vessels/job_launcher.py b/rslp/sentinel2_vessels/job_launcher.py
index 81f517a..c8b4cba 100644
--- a/rslp/sentinel2_vessels/job_launcher.py
+++ b/rslp/sentinel2_vessels/job_launcher.py
@@ -25,6 +25,10 @@
 
 from .predict_pipeline import PredictionTask
 
+CLUSTERS = [
+    "ai2/jupiter-cirrascale-2",
+]
+
 
 def launch_job(image_name: str, tasks: list[PredictionTask]) -> None:
     """Launch job for the Sentinel-2 scene.
@@ -60,9 +64,7 @@ def launch_job(image_name: str, tasks: list[PredictionTask]) -> None:
                 "/tmp/x/",
             ],
             constraints=Constraints(
-                cluster=[
-                    "ai2/jupiter-cirrascale-2",
-                ]
+                cluster=CLUSTERS,
             ),
             preemptible=True,
             datasets=[