Fix failing test with race condition (#32140)
* Fix failing test with race condition * make fixup * monotonic_ns instead of randint * uuid4 instead of monotonic_ns * Add a finally cleanup step
This commit is contained in:
@@ -20,6 +20,7 @@ import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
from shutil import copyfile
|
||||
from uuid import uuid4
|
||||
|
||||
from huggingface_hub import HfFolder, Repository, create_repo, delete_repo
|
||||
from requests.exceptions import HTTPError
|
||||
@@ -431,9 +432,11 @@ class ProcessorPushToHubTester(unittest.TestCase):
|
||||
|
||||
processor = CustomProcessor(feature_extractor, tokenizer)
|
||||
|
||||
random_repo_id = f"{USER}/test-dynamic-processor-{uuid4()}"
|
||||
try:
|
||||
with tempfile.TemporaryDirectory() as tmp_dir:
|
||||
create_repo(f"{USER}/test-dynamic-processor", token=self._token)
|
||||
repo = Repository(tmp_dir, clone_from=f"{USER}/test-dynamic-processor", token=self._token)
|
||||
create_repo(random_repo_id, token=self._token)
|
||||
repo = Repository(tmp_dir, clone_from=random_repo_id, token=self._token)
|
||||
processor.save_pretrained(tmp_dir)
|
||||
|
||||
# This has added the proper auto_map field to the feature extractor config
|
||||
@@ -463,6 +466,8 @@ class ProcessorPushToHubTester(unittest.TestCase):
|
||||
|
||||
repo.push_to_hub()
|
||||
|
||||
new_processor = AutoProcessor.from_pretrained(f"{USER}/test-dynamic-processor", trust_remote_code=True)
|
||||
new_processor = AutoProcessor.from_pretrained(random_repo_id, trust_remote_code=True)
|
||||
# Can't make an isinstance check because the new_processor is from the CustomProcessor class of a dynamic module
|
||||
self.assertEqual(new_processor.__class__.__name__, "CustomProcessor")
|
||||
finally:
|
||||
delete_repo(repo_id=random_repo_id)
|
||||
|
||||
Reference in New Issue
Block a user