[CI] Add missing packages for smoke test

ghstack-source-id: 6a78016ac15a324c036ebb85ab82efcbc8dc3fbd Pull Request resolved: #1149
pytorch · Dec 19, 2024 · b0890bc · b0890bc
1 parent b19fc05
commit b0890bc
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/.github/scripts/version_script.sh b/.github/scripts/version_script.sh
@@ -1,3 +1,5 @@
 #!/bin/bash
 
 export TENSORDICT_BUILD_VERSION=0.6.2
+
+python -m pip install packaging
diff --git a/.github/workflows/build-wheels-aarch64-linux.yml b/.github/workflows/build-wheels-aarch64-linux.yml
@@ -33,7 +33,6 @@ jobs:
       matrix:
         include:
           - repository: pytorch/tensordict
-            smoke-test-script: test/smoke_test.py
             package-name: tensordict
     name: pytorch/tensordict
     uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main

diff --git a/.github/workflows/build-wheels-linux.yml b/.github/workflows/build-wheels-linux.yml
@@ -33,6 +33,7 @@ jobs:
         include:
           - repository: pytorch/tensordict
             smoke-test-script: test/smoke_test.py
+            env-script: .github/scripts/install-deps-smoke-test.sh
             package-name: tensordict
     name: pytorch/tensordict
     uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main

diff --git a/.github/workflows/build-wheels-m1.yml b/.github/workflows/build-wheels-m1.yml
@@ -33,6 +33,7 @@ jobs:
         include:
           - repository: pytorch/tensordict
             smoke-test-script: test/smoke_test.py
+            env-script: .github/scripts/install-deps-smoke-test.sh
             package-name: tensordict
     name: pytorch/tensordict
     uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@main