From 6a865fb34d75d95c9a95c37e3b5ce94fa9784af2 Mon Sep 17 00:00:00 2001
From: Tim Saucer <timsaucer@gmail.com>
Date: Sat, 27 Jul 2024 09:05:41 -0400
Subject: [PATCH] Add exports for object store to match internal

---
 python/datafusion/object_store.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/python/datafusion/object_store.py b/python/datafusion/object_store.py
index a9bb83d29..c927e7614 100644
--- 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)