Remove datasets requirement (#14795)
This commit is contained in:
@@ -31,7 +31,6 @@ from transformers import logging as transformers_logging
|
||||
|
||||
from .deepspeed import is_deepspeed_available
|
||||
from .file_utils import (
|
||||
is_datasets_available,
|
||||
is_detectron2_available,
|
||||
is_faiss_available,
|
||||
is_flax_available,
|
||||
@@ -513,15 +512,6 @@ def require_torch_tf32(test_case):
|
||||
return test_case
|
||||
|
||||
|
||||
def require_datasets(test_case):
|
||||
"""Decorator marking a test that requires datasets."""
|
||||
|
||||
if not is_datasets_available():
|
||||
return unittest.skip("test requires `datasets`")(test_case)
|
||||
else:
|
||||
return test_case
|
||||
|
||||
|
||||
def require_detectron2(test_case):
|
||||
"""Decorator marking a test that requires detectron2."""
|
||||
if not is_detectron2_available():
|
||||
|
||||
Reference in New Issue
Block a user