reset accelerate env variables after each test (huggingface#24107)

novice03 · Jun 23, 2023 · 2e27e48 · 2e27e48
1 parent 236db9e
commit 2e27e48
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/transformers/testing_utils.py b/src/transformers/testing_utils.py
@@ -1339,6 +1339,11 @@ def tearDown(self):
             AcceleratorState._reset_state()
             PartialState._reset_state()
 
+            # delete all the env variables having `ACCELERATE` in them
+            for k in list(os.environ.keys()):
+                if "ACCELERATE" in k:
+                    del os.environ[k]
+
 
 def mockenv(**kwargs):
     """