Lock tensordict to 0.2.1

vladmandic · Feb 7, 2024 · 376bace · 376bace
1 parent 902907a
commit 376bace
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/installer.py b/installer.py
@@ -461,11 +461,11 @@ def check_torch():
                 install(os.environ.get('MKL_PACKAGE', 'mkl==2024.0.0'), 'mkl')
                 install(os.environ.get('DPCPP_PACKAGE', 'mkl-dpcpp==2024.0.0'), 'mkl-dpcpp')
         else:
-            if sys.version_info[1] == 11:
+            if sys.version_info.minor == 11:
                 pytorch_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torch-2.1.0a0+cxx11.abi-cp311-cp311-win_amd64.whl'
                 torchvision_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torchvision-0.16.0a0+cxx11.abi-cp311-cp311-win_amd64.whl'
                 ipex_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/intel_extension_for_pytorch-2.1.10+xpu-cp311-cp311-win_amd64.whl'
-            elif sys.version_info[1] == 10:
+            elif sys.version_info.minor == 10:
                 pytorch_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torch-2.1.0a0+cxx11.abi-cp310-cp310-win_amd64.whl'
                 torchvision_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torchvision-0.16.0a0+cxx11.abi-cp310-cp310-win_amd64.whl'
                 ipex_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/intel_extension_for_pytorch-2.1.10+xpu-cp310-cp310-win_amd64.whl'

diff --git a/requirements.txt b/requirements.txt
@@ -34,7 +34,7 @@ rich
 safetensors
 scipy
 tb_nightly
-tensordict
+tensordict==0.2.1
 toml
 torchdiffeq
 voluptuous