reset accelerate env variables after each test (#24107)

huggingface · Jun 8, 2023 · 17db177 · 17db177
1 parent 905892f
commit 17db177
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):
     """