diff --git a/requirements/fabric/test.txt b/requirements/fabric/test.txt index c4e1cf8866..26395bbdb9 100644 --- a/requirements/fabric/test.txt +++ b/requirements/fabric/test.txt @@ -4,4 +4,4 @@ pytest==7.2.0 pytest-cov==4.0.0 pre-commit==2.20.0 click==8.1.3 -tensorboard>=2.9.1, <2.12.0 +tensorboardX>=2.2, <=2.5.1 # min version is set by torch.onnx missing attribute diff --git a/tests/tests_fabric/conftest.py b/tests/tests_fabric/conftest.py index bef3b42ede..3df05bc56e 100644 --- a/tests/tests_fabric/conftest.py +++ b/tests/tests_fabric/conftest.py @@ -54,6 +54,7 @@ def restore_env_variables(): "RANK", # set by DeepSpeed "POPLAR_ENGINE_OPTIONS", # set by IPUStrategy "CUDA_MODULE_LOADING", # leaked since PyTorch 1.13 + "CRC32C_SW_MODE", # set by tensorboardX } leaked_vars.difference_update(allowlist) assert not leaked_vars, f"test is leaking environment variable(s): {set(leaked_vars)}"