Merge pull request #298 from allenai/yawenz/20260317_pass_checkpoint_for_oe

Fix unexpanded RSLP_PREFIX in management_dir

Author: yawenzzzz

rslp/olmoearth_evals/20260413_evals.md rslp/olmoearth_evals/20260415_evals.mdrslp/olmoearth_evals/20260413_evals.md renamed to rslp/olmoearth_evals/20260415_evals.md

@@ -1,8 +1,8 @@
-# 20260413 Evals
+# 20260415 Evals
 
-Image: `yawenzzzz/rslpomp20260413f`
+Image: `yawenzzzz/rslpomp20260415a`
 Checkpoint: `/weka/dfive-default/olmoearth_pretrain/checkpoints/yawenzzzz/single_bandset_no_s1_drop_random_time_dropout_0.2/step600000`
-W&B project: `2026_04_12_v1.1_finetune_partner_tasks`
+W&B project: `2026_04_15_v1.1_finetune_partner_tasks`
 
 Note: all tasks use `use_legacy_timestamps: true` since the checkpoint was trained with legacy timestamps.
 
@@ -12,9 +12,9 @@ Note: all tasks use `use_legacy_timestamps: true` since the checkpoint was train
 python -m rslp.main olmoearth_evals launch \
     --models '["olmoearth"]' \
     --tasks '["forest_loss_driver", "ecosystem", "lfmc_ts", "mangrove_ts", "marine_infra_ts", "nandi_ts", "sentinel2_vessels", "sentinel2_vessel_length", "sentinel2_vessel_type", "solar_farm_ts", "wind_turbine_ts", "awf_ts"]' \
-    --prefix 20260413 \
-    --image_name yawenzzzz/rslpomp20260413f \
-    --project 2026_04_12_v1.1_finetune_partner_tasks \
+    --prefix 20260415 \
+    --image_name yawenzzzz/rslpomp20260415a \
+    --project 2026_04_15_v1.1_finetune_partner_tasks \
     --model_config '{"use_legacy_timestamps": "true", "checkpoint_path": "/weka/dfive-default/olmoearth_pretrain/checkpoints/yawenzzzz/single_bandset_no_s1_drop_random_time_dropout_0.2/step600000"}'
 ```
 
@@ -24,9 +24,9 @@ python -m rslp.main olmoearth_evals launch \
 python -m rslp.main olmoearth_evals launch \
     --models '["olmoearth"]' \
     --tasks '["awf_mm", "lfmc_mm", "mangrove_mm", "marine_infra_mm", "nandi_mm", "solar_farm_mm", "wind_turbine_mm"]' \
-    --prefix 20260413 \
-    --image_name yawenzzzz/rslpomp20260413f \
-    --project 2026_04_12_v1.1_finetune_partner_tasks \
+    --prefix 20260415 \
+    --image_name yawenzzzz/rslpomp20260415a \
+    --project 2026_04_15_v1.1_finetune_partner_tasks \
     --model_config '{"use_legacy_timestamps": "true", "checkpoint_path": "/weka/dfive-default/olmoearth_pretrain/checkpoints/yawenzzzz/single_bandset_no_s1_drop_random_time_dropout_0.2/step600000"}'
 ```
 
@@ -36,9 +36,9 @@ python -m rslp.main olmoearth_evals launch \
 python -m rslp.main olmoearth_evals launch \
     --models '["olmoearth"]' \
     --tasks '["sentinel1_vessels"]' \
-    --prefix 20260413 \
-    --image_name yawenzzzz/rslpomp20260413f \
-    --project 2026_04_12_v1.1_finetune_partner_tasks \
+    --prefix 20260415 \
+    --image_name yawenzzzz/rslpomp20260415a \
+    --project 2026_04_15_v1.1_finetune_partner_tasks \
     --model_config '{"use_legacy_timestamps": "true", "checkpoint_path": "/weka/dfive-default/olmoearth_pretrain/checkpoints/yawenzzzz/single_bandset_no_s1_drop_random_time_dropout_0.2/step600000"}'
 ```
 
@@ -48,8 +48,8 @@ python -m rslp.main olmoearth_evals launch \
 python -m rslp.main olmoearth_evals launch \
     --models '["olmoearth"]' \
     --tasks '["landsat_vessels"]' \
-    --prefix 20260413 \
-    --image_name yawenzzzz/rslpomp20260413f \
-    --project 2026_04_12_v1.1_finetune_partner_tasks \
+    --prefix 20260415 \
+    --image_name yawenzzzz/rslpomp20260415a \
+    --project 2026_04_15_v1.1_finetune_partner_tasks \
     --model_config '{"use_legacy_timestamps": "true", "checkpoint_path": "/weka/dfive-default/olmoearth_pretrain/checkpoints/yawenzzzz/single_bandset_no_s1_drop_random_time_dropout_0.2/step600000"}'
 ```

rslp/olmoearth_evals/launch.py

@@ -1,6 +1,7 @@
 """Launch OlmoEarth fine-tuning evals."""
 
 import json
+import os
 import subprocess  # nosec
 
 from rslearn.utils.geometry import PixelBounds
@@ -147,8 +148,11 @@ def launch(
                 ]
 
             # Build extra_args for the training script.
+            # Expand RSLP_PREFIX here since rslearn's RslearnArgumentParser
+            # only substitutes env vars in config file content, not CLI args.
+            rslp_prefix = os.environ["RSLP_PREFIX"]
             all_extra_args: list[str] = [
-                "--management_dir=${RSLP_PREFIX}/projects",
+                f"--management_dir={rslp_prefix}/projects",
             ]
             if use_embeddings:
                 all_extra_args.append(