diff --git a/tensorflow_datasets/core/community/huggingface_wrapper.py b/tensorflow_datasets/core/community/huggingface_wrapper.py index 5d9bca20b5e..9e8fc46a487 100644 --- a/tensorflow_datasets/core/community/huggingface_wrapper.py +++ b/tensorflow_datasets/core/community/huggingface_wrapper.py @@ -27,6 +27,7 @@ from typing import Iterator, NamedTuple, Optional, Union from unittest import mock +from importlib.machinery import ModuleSpec from etils import epath from tensorflow_datasets.core import dataset_builder from tensorflow_datasets.core import dataset_info @@ -248,7 +249,7 @@ class _MockedHFDatasets(types.ModuleType): def __init__(self): super().__init__('datasets') - + self.__spec__ = ModuleSpec('datasets', None) # pylint: disable=invalid-name Version = Version