Skip to content

Commit 257b992

Browse files
authored
Refactor test runs to use environment variable for token
1 parent 6d61e21 commit 257b992

File tree

1 file changed

+18
-6
lines changed

1 file changed

+18
-6
lines changed

.github/workflows/slow_tests_gaudi3.yml

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,9 @@ jobs:
5050
- name: Checkout
5151
uses: actions/checkout@v6
5252
- name: Run tests
53-
run: /bin/bash tests/ci/slow_tests_diffusers.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
53+
env:
54+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
55+
run: /bin/bash tests/ci/slow_tests_diffusers.sh "$TEXT_GENERATION_CI_HUB_TOKEN"
5456

5557
deepspeed:
5658
name: Test DeepSpeed models
@@ -73,7 +75,9 @@ jobs:
7375
- name: Checkout
7476
uses: actions/checkout@v6
7577
- name: Run tests
76-
run: /bin/bash tests/ci/slow_tests_deepspeed.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
78+
env:
79+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
80+
run: /bin/bash tests/ci/slow_tests_deepspeed.sh "$TEXT_GENERATION_CI_HUB_TOKEN"
7781

7882
fsdp:
7983
name: Test FSDP models
@@ -96,7 +100,9 @@ jobs:
96100
- name: Checkout
97101
uses: actions/checkout@v6
98102
- name: Run tests
99-
run: make slow_tests_fsdp TOKEN=${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
103+
env:
104+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
105+
run: make slow_tests_fsdp TOKEN="$TEXT_GENERATION_CI_HUB_TOKEN"
100106

101107
multi-card:
102108
name: Test multi-card models
@@ -119,7 +125,9 @@ jobs:
119125
- name: Checkout
120126
uses: actions/checkout@v6
121127
- name: Run tests
122-
run: /bin/bash tests/ci/slow_tests_8x.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
128+
env:
129+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
130+
run: /bin/bash tests/ci/slow_tests_8x.sh "$TEXT_GENERATION_CI_HUB_TOKEN"
123131

124132
single-card:
125133
name: Test single-card models
@@ -143,7 +151,9 @@ jobs:
143151
- name: Checkout
144152
uses: actions/checkout@v6
145153
- name: Run tests
146-
run: /bin/bash tests/ci/slow_tests_1x.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
154+
env:
155+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
156+
run: /bin/bash tests/ci/slow_tests_1x.sh "$TEXT_GENERATION_CI_HUB_TOKEN"
147157

148158
text-generation:
149159
name: Test text-generation example
@@ -168,7 +178,9 @@ jobs:
168178
- name: Checkout
169179
uses: actions/checkout@v6
170180
- name: Run tests
171-
run: make slow_tests_text_generation_example TOKEN=${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
181+
env:
182+
TEXT_GENERATION_CI_HUB_TOKEN: ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
183+
run: make slow_tests_text_generation_example TOKEN="$TEXT_GENERATION_CI_HUB_TOKEN"
172184

173185
trl:
174186
name: Test TRL integration

0 commit comments

Comments
 (0)