[CI] Fix linter, test, and docs CI failures (#1572)

Author: vmoens

.github/unittest/linux/scripts/setup_env.sh

@@ -94,8 +94,10 @@ if [ "${PYTHON_VERSION}" == "3.14t" ]; then
     # Install test dependencies (mirrors environment.yml)
     pip install pybind11 numpy expecttest pyyaml hypothesis future cloudpickle \
         pytest pytest-benchmark pytest-cov pytest-mock pytest-instafail \
-        pytest-rerunfailures pytest-timeout coverage h5py orjson ninja protobuf redis
-    # Note: mosaicml-streaming may not be available for 3.14t yet, skip if fails
+        pytest-rerunfailures pytest-timeout coverage ninja protobuf redis
+    # h5py, orjson and mosaicml-streaming may not be available for 3.14t yet
+    pip install h5py || echo "h5py not available for Python 3.14t, skipping"
+    pip install orjson || echo "orjson not available for Python 3.14t, skipping"
     pip install mosaicml-streaming || echo "mosaicml-streaming not available for Python 3.14t, skipping"
 else
     # For regular Python, use conda

.github/workflows/docs.yml

@@ -55,8 +55,8 @@ jobs:
         eval "$(./conda/bin/conda shell.bash hook)"
         conda activate "./env"
 
-        # Install PyTorch (CPU version for docs)
-        python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U --quiet --root-user-action=ignore
+        # Install PyTorch and torchvision (CPU version for docs)
+        python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U --quiet --root-user-action=ignore
 
         # Install tensordict runtime deps explicitly (except torch which is handled above),
         # then install tensordict without resolving dependencies to avoid any solver changing the

benchmarks/storage/bench_redis.py

@@ -16,7 +16,6 @@
 import time
 
 import torch
-
 from tensordict import TensorDict
 
 _has_redis_pkg = importlib.util.find_spec("redis", None) is not None

tensordict/redis.py

@@ -17,7 +17,6 @@
 from typing import Any, Callable, Tuple, Type, TYPE_CHECKING
 
 import torch
-
 from tensordict._td import (
     _TensorDictKeysView,
     _unravel_key_to_tuple,

test/test_h5.py

@@ -100,6 +100,7 @@ def test_h5_nontensor(self, tmpdir):
         assert is_non_tensor(td_recover.get(("g", "h")))
 
 
+@pytest.mark.skipif(not _has_h5py, reason="h5py not found.")
 def test_auto_batch_size(tmpdir):
     tmpdir = Path(tmpdir)
     td = TensorDict(

test/test_setup.py

@@ -4,6 +4,7 @@
 import shutil
 import subprocess
 import sys
+import sysconfig
 import venv
 from pathlib import Path
 
@@ -102,6 +103,10 @@ def test_install_strategies_version(
 ):
     if shutil.which("cmake") is None:
         pytest.skip("cmake not available")
+    if no_build_isolation and sysconfig.get_config_var("Py_GIL_DISABLED"):
+        pytest.skip(
+            "no-build-isolation installs are unreliable on free-threaded Python"
+        )
 
     base_version = (_ROOT / "version.txt").read_text().strip()
     if not base_version:

test/test_tensorclass.py

@@ -67,6 +67,7 @@
     pytest.mark.filterwarnings(
         "ignore:You are using `torch.load` with `weights_only=False`"
     ),
+    pytest.mark.filterwarnings("ignore::pytest.PytestUnraisableExceptionWarning"),
 ]
 
 IS_FB = os.getenv("PYTORCH_TEST_FBCODE")

test/test_tensordict.py

@@ -159,6 +159,7 @@
     pytest.mark.filterwarnings(
         "ignore:No PYTORCH_KERNEL_CACHE_PATH or HOME environment variable set"
     ),
+    pytest.mark.filterwarnings("ignore::pytest.PytestUnraisableExceptionWarning"),
 ]
 
 mp_ctx = "spawn"
@@ -12795,6 +12796,8 @@ def test_map_seed_single(self):
     @pytest.mark.parametrize("h5", [False, True])
     @pytest.mark.parametrize("has_out", [False, True])
     def test_index_with_generator(self, chunksize, num_chunks, h5, has_out, tmpdir):
+        if h5 and not _has_h5py:
+            pytest.skip("h5py not installed")
         gc.collect()
         input = TensorDict({"a": torch.arange(10), "b": torch.arange(10)}, [10])
         if h5: