diff --git a/pystreamapi/loaders/__init__.py b/pystreamapi/loaders/__init__.py index f12d129..11b171d 100644 --- a/pystreamapi/loaders/__init__.py +++ b/pystreamapi/loaders/__init__.py @@ -1,11 +1,18 @@ from pystreamapi.loaders.__csv.__csv_loader import csv from pystreamapi.loaders.__json.__json_loader import json -from pystreamapi.loaders.__xml.__xml_loader import xml -from pystreamapi.loaders.__yaml.__yaml_loader import yaml - -__all__ = [ - 'csv', - 'json', - 'xml', - 'yaml' -] + +__all__ = ['csv', 'json'] + +try: + from pystreamapi.loaders.__xml.__xml_loader import xml + + __all__.append('xml') +except ImportError: + ... + +try: + from pystreamapi.loaders.__yaml.__yaml_loader import yaml + + __all__.append('yaml') +except ImportError: + ...