Skip to content

Commit 442fc6a

Browse files
committed
canary: pin wandb namespace for profile summarize
1 parent 9595666 commit 442fc6a

2 files changed

Lines changed: 14 additions & 2 deletions

File tree

.github/workflows/marin-canary-ferry-cw.yaml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ jobs:
1818
cancel-in-progress: true
1919
env:
2020
RUN_ID: canary-gpu-${{ github.run_id }}-${{ github.run_attempt }}
21+
WANDB_ENTITY: marin-community
22+
WANDB_PROJECT: marin
2123

2224
steps:
2325
- name: Checkout code
@@ -64,6 +66,8 @@ jobs:
6466
--memory=16G --disk=16G --cpu=1 --extra=cpu \
6567
-e MARIN_PREFIX s3://marin-us-west-04a/marin/ \
6668
-e RUN_ID "$RUN_ID" \
69+
-e WANDB_ENTITY "$WANDB_ENTITY" \
70+
-e WANDB_PROJECT "$WANDB_PROJECT" \
6771
-e WANDB_API_KEY "$WANDB_API_KEY" \
6872
-e HF_TOKEN "$HF_TOKEN" \
6973
-- python -m experiments.ferries.canary_ferry_cw
@@ -79,7 +83,9 @@ jobs:
7983
shell: bash -l {0}
8084
run: |
8185
.venv/bin/python -m marin.profiling.cli summarize \
82-
--run-target "marin-community/marin/$RUN_ID" \
86+
--run-target "$RUN_ID" \
87+
--entity "$WANDB_ENTITY" \
88+
--project "$WANDB_PROJECT" \
8389
--output profile_summary.json
8490
.venv/bin/python -m marin.profiling.cli report \
8591
--summary profile_summary.json \

.github/workflows/marin-canary-ferry.yaml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@ jobs:
1414
cancel-in-progress: true
1515
env:
1616
RUN_ID: canary-tpu-${{ github.run_id }}-${{ github.run_attempt }}
17+
WANDB_ENTITY: marin-community
18+
WANDB_PROJECT: marin
1719

1820
steps:
1921
- name: Checkout code
@@ -56,6 +58,8 @@ jobs:
5658
.venv/bin/python lib/marin/src/marin/run/ray_run.py \
5759
--cluster us-central1 \
5860
-e RUN_ID "$RUN_ID" \
61+
-e WANDB_ENTITY "$WANDB_ENTITY" \
62+
-e WANDB_PROJECT "$WANDB_PROJECT" \
5963
-- python experiments/ferries/canary_ferry.py
6064
env:
6165
WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
@@ -71,7 +75,9 @@ jobs:
7175
shell: bash -l {0}
7276
run: |
7377
.venv/bin/python -m marin.profiling.cli summarize \
74-
--run-target "marin-community/marin/$RUN_ID" \
78+
--run-target "$RUN_ID" \
79+
--entity "$WANDB_ENTITY" \
80+
--project "$WANDB_PROJECT" \
7581
--output profile_summary.json
7682
.venv/bin/python -m marin.profiling.cli report \
7783
--summary profile_summary.json \

0 commit comments

Comments
 (0)