Update

[ghstack-poisoned]
pytorch · Dec 19, 2024 · abbb69c · abbb69c
1 parent bd47ee1
commit abbb69c
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/.github/scripts/install-deps-smoke-test.sh b/.github/scripts/install-deps-smoke-test.sh
@@ -0,0 +1 @@
+python3 -m pip install packaging
diff --git a/.github/workflows/build-wheels-aarch64-linux.yml b/.github/workflows/build-wheels-aarch64-linux.yml
@@ -34,6 +34,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-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