Skip to content

Commit 17db177

Browse files
pacman100sgugger
authored andcommitted
reset accelerate env variables after each test (#24107)
1 parent 905892f commit 17db177

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/transformers/testing_utils.py

+5
Original file line numberDiff line numberDiff line change
@@ -1339,6 +1339,11 @@ def tearDown(self):
13391339
AcceleratorState._reset_state()
13401340
PartialState._reset_state()
13411341

1342+
# delete all the env variables having `ACCELERATE` in them
1343+
for k in list(os.environ.keys()):
1344+
if "ACCELERATE" in k:
1345+
del os.environ[k]
1346+
13421347

13431348
def mockenv(**kwargs):
13441349
"""

0 commit comments

Comments
 (0)