[0.28.0-dlc] fix the integration test to build on staging for gpu (#2313

)
deepjavalibrary · Aug 27, 2024 · 9fe4d68 · 9fe4d68
1 parent bf083ae
commit 9fe4d68
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
@@ -168,7 +168,7 @@ jobs:
             ${{ runner.os }}-gradle-
       - name: Install DJL-Bench
         working-directory: benchmark
-        run: ./gradlew installOnLinux
+        run: ./gradlew installOnLinux  -Pstaging
       - name: Build container name
         run: ./serving/docker/scripts/docker_name_builder.sh pytorch-gpu ${{ github.event.inputs.djl-version }}
       - name: Download models and dockers
@@ -226,7 +226,7 @@ jobs:
             ${{ runner.os }}-gradle-
       - name: Install DJL-Bench
         working-directory: benchmark
-        run: ./gradlew installOnLinux
+        run: ./gradlew installOnLinux -Pstaging
       - name: Build container name
         run: ./serving/docker/scripts/docker_name_builder.sh aarch64 ${{ github.event.inputs.djl-version }}
       - name: Download models and dockers