Fix temporary directory cleanup in test_callbacks.py

huggingface · Nov 21, 2024 · a57d88a · a57d88a
1 parent 7fe26d5
commit a57d88a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_callbacks.py b/tests/test_callbacks.py
@@ -279,7 +279,7 @@ def setUp(self):
         self.dataset = load_dataset("trl-internal-testing/zen", "standard_preference", split="train")
 
     def test_callback(self):
-        with tempfile.TemporaryDirectory() as tmp_dir:
+        with tempfile.TemporaryDirectory(ignore_cleanup_errors=True) as tmp_dir:
             training_args = DPOConfig(
                 output_dir=tmp_dir,
                 num_train_epochs=1,
@@ -305,7 +305,7 @@ def test_callback(self):
             pass
 
     def test_every_checkpoint(self):
-        with tempfile.TemporaryDirectory() as tmp_dir:
+        with tempfile.TemporaryDirectory(ignore_cleanup_errors=True) as tmp_dir:
             training_args = DPOConfig(
                 output_dir=tmp_dir,
                 num_train_epochs=1,