add CLUSTERS constant

allenai · Feb 6, 2025 · fdc56b3 · fdc56b3
1 parent 723b9c2
commit fdc56b3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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=[