From 0f2360509449cb16362b2e316dcab9fb4d1ee7f0 Mon Sep 17 00:00:00 2001
From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
Date: Thu, 8 Jun 2023 18:49:07 +0530
Subject: [PATCH] reset accelerate env variables after each test (#24107)

---
 src/transformers/testing_utils.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/transformers/testing_utils.py b/src/transformers/testing_utils.py
index 40d5d4c022..0a35494765 100644
--- a/src/transformers/testing_utils.py
+++ b/src/transformers/testing_utils.py
@@ -1339,6 +1339,11 @@ class TestCasePlus(unittest.TestCase):
             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):
     """