fabric: test with tbX (#16511)

This commit is contained in:
Jirka Borovec 2023-01-26 21:52:02 +09:00 committed by GitHub
parent d871f70389
commit 50fd12f841
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions

View File

@ -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

View File

@ -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)}"