force install of torch / torch-xla in test_installs

huggingface · Apr 15, 2024 · 51e1cc6 · 51e1cc6
1 parent ed16308
commit 51e1cc6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Makefile b/Makefile
@@ -69,6 +69,7 @@ pypi_upload: ${PACKAGE_DIST} ${PACKAGE_WHEEL}
 # Tests
 test_installs:
 	python -m pip uninstall -y torchvision torchaudio
+	python -m pip install torch~=2.2.0 torch_xla[tpu]~=2.2.0 -f https://storage.googleapis.com/libtpu-releases/index.html
 	python -m pip install .[tests]
 
 tests: test_installs