fix(opendataset): fix the AttributeError calling mocker class methods

PR Closed: #1119
Graviti-AI · Nov 22, 2021 · 2dab64b · 2dab64b
1 parent 5771933
commit 2dab64b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tensorbay/opendataset/_utility/mocker.py b/tensorbay/opendataset/_utility/mocker.py
@@ -7,15 +7,19 @@
 from tensorbay.exception import ModuleImportError
 
 
-class Image:
+class ImageMocker:
     """Raise import PIL error for data loader."""
 
     def __getattribute__(self, name: str) -> None:
         raise ModuleImportError(module_name="pillow")
 
 
-class xmltodict:
+class xmltodictMocker:
     """Raise import xmltodict error for data loader."""
 
     def __getattribute__(self, name: str) -> None:
         raise ModuleImportError(module_name="xmltodict")
+
+
+Image = ImageMocker()
+xmltodict = xmltodictMocker()