From 50fd12f841bc10ec093e80d0acf3e3b51cd927fe Mon Sep 17 00:00:00 2001 From: Jirka Borovec <6035284+Borda@users.noreply.github.com> Date: Thu, 26 Jan 2023 21:52:02 +0900 Subject: [PATCH] fabric: test with tbX (#16511) --- requirements/fabric/test.txt | 2 +- tests/tests_fabric/conftest.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) 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)}"