Update Beaker budget (ai2/es-platform -> ai2/atec-olmoearth)

docs/Setup-Internal.md

@@ -43,7 +43,7 @@ Create a GitHub token that can clone this repo on Beaker. Generate a token [here
 
 ```bash
 beaker config set default_workspace ai2/earth-systems
-beaker workspace set-budget ai2/earth-systems ai2/es-platform
+beaker workspace set-budget ai2/earth-systems ai2/atec-olmoearth
 ```
 
 ### 3. Set Beaker Secrets
@@ -144,7 +144,7 @@ Evaluation datasets have default paths configured in [`olmoearth_pretrain/evals/
 ## Beaker Information
 
 **Quick Reference:**
-- **Budget:** `ai2/es-platform`
+- **Budget:** `ai2/atec-olmoearth`
 - **Workspace:** `ai2/earth-systems`
 - **Weka:** `weka://dfive-default`
 

olmoearth_pretrain/dataset_creation/internal_docs.md

@@ -4,7 +4,7 @@ A Beaker session can be used to run most of the window creation and data
 materialization steps:
 
 ```
-beaker session create --budget ai2/es-platform --workspace ai2/earth-systems --priority high --gpus 1 --shared-memory 128GiB --bare --mount src=weka,ref=dfive-default,dst=/weka/dfive-default
+beaker session create --budget ai2/atec-olmoearth --workspace ai2/earth-systems --priority high --gpus 1 --shared-memory 128GiB --bare --mount src=weka,ref=dfive-default,dst=/weka/dfive-default
 ```
 
 The only exception is for Sentinel-1 and Sentinel-2 L2A, where it may be desirable to

olmoearth_pretrain/inference_benchmarking/constants.py

@@ -1,7 +1,7 @@
 """Shared constants important for inference throughput benchmarking."""
 
 # BEAKER-LAND
-BEAKER_BUDGET = "ai2/es-platform"
+BEAKER_BUDGET = "ai2/atec-olmoearth"
 BEAKER_WORKSPACE = "ai2/earth-systems"
 WEKA_BUCKET = "dfive-default"
 BEAKER_TASK_PRIORITY = "normal"

olmoearth_pretrain/internal/common.py

@@ -24,7 +24,7 @@
 )
 
 logger = logging.getLogger(__name__)
-BUDGET = "ai2/es-platform"
+BUDGET = "ai2/atec-olmoearth"
 WORKSPACE = "ai2/earth-systems"
 
 DEFAULT_OLMOEARTH_PRETRAIN_WEKA_BUCKET = BeakerWekaBucket(