fix

one_off_projects/convert_satlas_webmercator_to_rslearn/solar_farm/convert_siv_labels.py

@@ -77,7 +77,7 @@
         geometry["coordinates"] = (
             np.array(geometry["coordinates"]) - [window.bounds[0], window.bounds[1]]
         ).tolist()
-        shapes.append((geometry, 255))
+        shapes.append((geometry, 1))
     if shapes:
         mask = rasterio.features.rasterize(
             shapes,

rslp/common/README.md

@@ -24,4 +24,4 @@ Then you can launch the worker. To test on one machine:
 
 And to launch 100 workers on Beaker:
 
-    python -m rslp.main common launch skylight-proto-1 rslp-job-queue-YOURNAME-sub 100 --gpus 1 --shared_memory 256GiB
+    python -m rslp.main common launch BEAKER_IMAGE_NAME skylight-proto-1 rslp-job-queue-YOURNAME-sub 100 --gpus 1 --shared_memory 256GiB

rslp/common/worker.py

@@ -19,7 +19,8 @@
 )
 from google.cloud import pubsub_v1, storage
 
-from rslp.launch_beaker import BUDGET, DEFAULT_WORKSPACE, IMAGE_NAME, get_base_env_vars
+from rslp.launch_beaker import BUDGET, DEFAULT_WORKSPACE
+from rslp.launcher_lib import get_base_env_vars
 from rslp.log_utils import get_logger
 from rslp.main import run_workflow
 
@@ -196,6 +197,7 @@ def launch_worker(project_id: str, subscription_id: str) -> None:
 
 
 def launch_workers(
+    image_name: str,
     project_id: str,
     subscription_id: str,
     num_workers: int,
@@ -207,6 +209,7 @@ def launch_workers(
     """Start workers for the prediction jobs.
 
     Args:
+        image_name: the Beaker image name to use for the jobs.
         project_id: the Google Cloud project ID.
         subscription_id: the Pub/Sub subscription ID.
         num_workers: number of workers to launch
@@ -224,7 +227,7 @@ def launch_workers(
             spec = ExperimentSpec.new(
                 budget=BUDGET,
                 description="worker",
-                beaker_image=IMAGE_NAME,
+                beaker_image=image_name,
                 priority=priority,
                 command=["python", "-m", "rslp.main"],
                 arguments=[