From 2098c12f198b26d48ce304aeddf44210715e73d0 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Tue, 23 Jul 2024 08:18:12 +0000 Subject: [PATCH] avoid deadlocks --- tests/models/auto/test_processor_auto.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/auto/test_processor_auto.py b/tests/models/auto/test_processor_auto.py index 248e63d44483..a9bbe50af3a8 100644 --- a/tests/models/auto/test_processor_auto.py +++ b/tests/models/auto/test_processor_auto.py @@ -435,7 +435,8 @@ def test_push_to_hub_dynamic_processor(self): random_repo_id = f"{USER}/test-dynamic-processor-{uuid4()}" try: with tempfile.TemporaryDirectory() as tmp_dir: - create_repo(random_repo_id, token=self._token) + # exist_ok=True to avoid unlikely deadlocks + create_repo(random_repo_id, token=self._token, exist_ok=True) repo = Repository(tmp_dir, clone_from=random_repo_id, token=self._token) processor.save_pretrained(tmp_dir)