mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

JaxTestCase: prevent leakage of global state

Browse Source
This commit is contained in:
Jake VanderPlas 2022-01-26 15:49:59 -08:00
parent 18a661497c
commit 2afea82908
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/test_util.py
View File

@ -929,7 +929,7 @@ class JaxTestCase(parameterized.TestCase):
super().setUp()
self._original_config = {}
for key, value in self._default_config.items():
self._original_config[key] = getattr(config, key)
self._original_config[key] = config._read(key)
config.update(key, value)
# We use the adler32 hash for two reasons.