🐛 Fix individual loader import fails

pickwicksoft · Nov 7, 2024 · ef6616d · ef6616d
1 parent 44a4f7f
commit ef6616d
Showing 1 changed file with 19 additions and 9 deletions.
diff --git a/pystreamapi/loaders/__init__.py b/pystreamapi/loaders/__init__.py
@@ -1,11 +1,21 @@
+__all__ = []
+
 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__.append('csv')
+__all__.append('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:
+    ...