Skip to content

Commit 6d61e21

Browse files
authored
Update environment variable usage in CI workflow
1 parent 1e71118 commit 6d61e21

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

.github/workflows/slow_tests_gaudi2.yml

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ jobs:
4545
run: |
4646
docker pull vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest
4747
- name: Run tests
48+
env:
49+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
4850
run: |
4951
docker run \
5052
--rm \
@@ -60,7 +62,7 @@ jobs:
6062
--net=host \
6163
--ipc=host \
6264
vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest \
63-
/bin/bash tests/ci/slow_tests_diffusers.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
65+
/bin/bash tests/ci/slow_tests_diffusers.sh "$TEXT_GENERATION_CI_HUB_TOKEN"
6466
deepspeed:
6567
name: Test DeepSpeed models
6668
if: ${{ !cancelled() && (success() || failure()) }}
@@ -74,6 +76,8 @@ jobs:
7476
run: |
7577
docker pull vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest
7678
- name: Run tests
79+
env:
80+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
7781
run: |
7882
docker run \
7983
--rm \
@@ -89,7 +93,7 @@ jobs:
8993
--net=host \
9094
--ipc=host \
9195
vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest \
92-
/bin/bash tests/ci/slow_tests_deepspeed.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
96+
/bin/bash tests/ci/slow_tests_deepspeed.sh "$TEXT_GENERATION_CI_HUB_TOKEN"
9397
fsdp:
9498
name: Test FSDP models
9599
if: ${{ !cancelled() && (success() || failure()) }}
@@ -103,6 +107,8 @@ jobs:
103107
run: |
104108
docker pull vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest
105109
- name: Run tests
110+
env:
111+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
106112
run: |
107113
docker run \
108114
--rm \
@@ -118,7 +124,7 @@ jobs:
118124
--net=host \
119125
--ipc=host \
120126
vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest \
121-
make slow_tests_fsdp TOKEN=${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
127+
make slow_tests_fsdp TOKEN="$TEXT_GENERATION_CI_HUB_TOKEN"
122128
multi-card:
123129
name: Test multi-card models
124130
if: ${{ !cancelled() && (success() || failure()) }}
@@ -132,6 +138,8 @@ jobs:
132138
run: |
133139
docker pull vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest
134140
- name: Run tests
141+
env:
142+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
135143
run: |
136144
docker run \
137145
--rm \
@@ -147,7 +155,7 @@ jobs:
147155
--net=host \
148156
--ipc=host \
149157
vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest \
150-
/bin/bash tests/ci/slow_tests_8x.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
158+
/bin/bash tests/ci/slow_tests_8x.sh "$TEXT_GENERATION_CI_HUB_TOKEN"
151159
single-card:
152160
name: Test single-card models
153161
if: ${{ !cancelled() && (success() || failure()) }}
@@ -194,6 +202,8 @@ jobs:
194202
run: |
195203
docker pull vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest
196204
- name: Run tests
205+
env:
206+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
197207
run: |
198208
docker run \
199209
--rm \
@@ -209,7 +219,7 @@ jobs:
209219
--net=host \
210220
--ipc=host \
211221
vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest \
212-
make slow_tests_text_generation_example TOKEN=${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
222+
make slow_tests_text_generation_example TOKEN="$TEXT_GENERATION_CI_HUB_TOKEN"
213223
trl:
214224
name: Test TRL integration
215225
if: ${{ !cancelled() && (success() || failure()) }}

0 commit comments

Comments
 (0)