From 42ad06604561466777661b49d3038b83a6def92f Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Thu, 19 Dec 2024 10:52:12 +0000
Subject: [PATCH] [CI] Add missing packages for smoke test

ghstack-source-id: 1ae2795ee734baac4419b722d4e11d522051b112
Pull Request resolved: https://github.com/pytorch/tensordict/pull/1149
---
 .github/workflows/build-wheels-aarch64-linux.yml | 1 -
 .github/workflows/build-wheels-linux.yml         | 1 +
 .github/workflows/build-wheels-m1.yml            | 1 +
 setup.py                                         | 8 +++++++-
 4 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build-wheels-aarch64-linux.yml b/.github/workflows/build-wheels-aarch64-linux.yml
index 6bd4fd1ac..9c690fcda 100644
--- 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
index b6fb57098..09afd984d 100644
--- 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
index 6609cb5ae..39cc8e19b 100644
--- 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
diff --git a/setup.py b/setup.py
index 26f2a321f..1a84252b6 100644
--- a/setup.py
+++ b/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",