[CI] Add missing packages for smoke test

ghstack-source-id: 1ae2795ee734baac4419b722d4e11d522051b112
Pull Request resolved: #1149

.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

.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

.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

setup.py

@@ -184,7 +184,13 @@ def _main(argv):
             "build_ext": BuildExtension.with_options(no_python_abi_suffix=True),
             "clean": clean,
         },
-        install_requires=[pytorch_package_dep, "numpy", "cloudpickle", "orjson"],
+        install_requires=[
+            pytorch_package_dep,
+            "numpy",
+            "cloudpickle",
+            "orjson",
+            "packaging",
+        ],
         extras_require={
             "tests": [
                 "pytest",