__init__.py 561 B

12345678910111213141516171819202122
  1. # Copyright (c) Alibaba, Inc. and its affiliates.
  2. from typing import TYPE_CHECKING
  3. from modelscope.utils.import_utils import LazyImportModule
  4. if TYPE_CHECKING:
  5. from .file import File, LocalStorage
  6. from .io import dump, dumps, load
  7. else:
  8. _import_structure = {
  9. 'io': ['dump', 'dumps', 'load'],
  10. 'file': ['File', 'LocalStorage']
  11. }
  12. import sys
  13. sys.modules[__name__] = LazyImportModule(
  14. __name__,
  15. globals()['__file__'],
  16. _import_structure,
  17. module_spec=__spec__,
  18. extra_objects={},
  19. )