Add exports for object store to match internal

apache · Jul 27, 2024 · 6a865fb · 6a865fb
1 parent 7695801
commit 6a865fb
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/python/datafusion/object_store.py b/python/datafusion/object_store.py
@@ -18,6 +18,18 @@
 
 from ._internal import object_store
 
+AmazonS3 = object_store.AmazonS3
+GoogleCloud = object_store.GoogleCloud
+LocalFileSystem = object_store.LocalFileSystem
+MicrosoftAzure = object_store.MicrosoftAzure
+
+__all__ = [
+    "AmazonS3",
+    "GoogleCloud",
+    "LocalFileSystem",
+    "MicrosoftAzure",
+]
+
 
 def __getattr__(name):
     return getattr(object_store, name)