Update (base update)

[ghstack-poisoned]

Author: Vincent Moens

.github/scripts/pre-build-script-win.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+pip install --upgrade setuptools
+
+export TORCHRL_BUILD_VERSION=0.7.0
+
+${CONDA_RUN} pip install git+https://github.com/pytorch/tensordict.git -U

.github/scripts/pre-build-script.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+pip install --upgrade setuptools
+
+${CONDA_RUN} pip install git+https://github.com/pytorch/tensordict.git -U

.github/scripts/td_script.sh

@@ -1,5 +1,25 @@
 #!/bin/bash
 
 export TORCHRL_BUILD_VERSION=0.7.0
+pip install --upgrade setuptools
 
-${CONDA_RUN} pip install git+https://github.com/pytorch/tensordict.git -U
+# Check if ARCH is set to aarch64
+ARCH=${ARCH:-}  # This sets ARCH to an empty string if it's not defined
+
+if pip list | grep -q torch; then
+    echo "Torch is installed."
+    if [[ "$ARCH" == "aarch64" ]]; then
+        ${CONDA_RUN} pip install git+https://github.com/pytorch/tensordict.git -U --no-deps
+    else
+        ${CONDA_RUN} pip install tensordict-nightly -U
+    fi
+elif [[ -n "${SMOKE_TEST_SCRIPT:-}" ]]; then
+    ${CONDA_RUN} ${PIP_INSTALL_TORCH}
+    if [[ "$ARCH" == "aarch64" ]]; then
+        ${CONDA_RUN} pip install git+https://github.com/pytorch/tensordict.git -U --no-deps
+    else
+        ${CONDA_RUN} pip install tensordict-nightly -U
+    fi
+else
+    echo "Torch is not installed - tensordict will be installed later."
+fi

.github/scripts/version_script.bat

@@ -36,6 +36,9 @@ if "%CU_VERSION%" == "xpu" call "C:\Program Files (x86)\Intel\oneAPI\setvars.bat
 
 set DISTUTILS_USE_SDK=1
 
+:: Upgrade setuptools before installing PyTorch
+pip install --upgrade setuptools==72.1.0 || exit /b 1
+
 set args=%1
 shift
 :start

.github/unittest/linux/scripts/environment.yml

@@ -17,6 +17,7 @@ dependencies:
     - pytest-instafail
     - pytest-rerunfailures
     - pytest-timeout
+    - pytest-asyncio
     - expecttest
     - pyyaml
     - scipy

.github/unittest/linux_distributed/scripts/environment.yml

@@ -16,6 +16,7 @@ dependencies:
     - pytest-mock
     - pytest-instafail
     - pytest-rerunfailures
+    - pytest-asyncio
     - expecttest
     - pyyaml
     - scipy

.github/unittest/linux_libs/scripts_ataridqn/environment.yml

@@ -15,6 +15,7 @@ dependencies:
     - pytest-instafail
     - pytest-rerunfailures
     - pytest-error-for-skips
+    - pytest-asyncio
     - expecttest
     - pyyaml
     - scipy

.github/unittest/linux_libs/scripts_brax/environment.yml

@@ -13,6 +13,7 @@ dependencies:
     - pytest-instafail
     - pytest-rerunfailures
     - pytest-error-for-skips
+    - pytest-asyncio
     - expecttest
     - pyyaml
     - scipy

.github/unittest/linux_libs/scripts_chess/environment.yml

@@ -13,6 +13,7 @@ dependencies:
     - pytest-instafail
     - pytest-rerunfailures
     - pytest-error-for-skips
+    - pytest-asyncio
     - expecttest
     - pyyaml
     - scipy

.github/unittest/linux_libs/scripts_d4rl/environment.yml

@@ -13,6 +13,7 @@ dependencies:
     - pytest-instafail
     - pytest-rerunfailures
     - pytest-error-for-skips
+    - pytest-asyncio
     - expecttest
     - pyyaml
     - scipy