disable caching

huggingface · May 24, 2024 · 16c6bd3 · 16c6bd3
1 parent 6d0c4b8
commit 16c6bd3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/utils/test_task_processors.py b/tests/utils/test_task_processors.py
@@ -19,7 +19,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Tuple, Union
 from unittest import TestCase
 
-from datasets import DatasetDict
+from datasets import DatasetDict, disable_caching
 from transformers import AutoConfig, AutoFeatureExtractor, AutoTokenizer
 
 from optimum.utils.preprocessing import TaskProcessorsManager
@@ -55,6 +55,8 @@
     },
 }
 
+disable_caching()  # trying to not saturate disk storage in CI
+
 
 # Taken from https://pynative.com/python-generate-random-string/
 def get_random_string(length: int) -> str: