fabric: test with tbX (#16511)
This commit is contained in:
parent
d871f70389
commit
50fd12f841
|
@ -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
|
||||
|
|
|
@ -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)}"
|
||||
|
|
Loading…
Reference in New Issue