diff --git a/python/datafusion/object_store.py b/python/datafusion/object_store.py index 3f51ad9d..7cc17506 100644 --- a/python/datafusion/object_store.py +++ b/python/datafusion/object_store.py @@ -25,7 +25,3 @@ Http = object_store.Http __all__ = ["AmazonS3", "GoogleCloud", "LocalFileSystem", "MicrosoftAzure", "Http"] - - -def __getattr__(name): - return getattr(object_store, name) diff --git a/python/datafusion/tests/test_sql.py b/python/datafusion/tests/test_sql.py index 575c11e5..681d80b4 100644 --- a/python/datafusion/tests/test_sql.py +++ b/python/datafusion/tests/test_sql.py @@ -22,7 +22,7 @@ from pyarrow.csv import write_csv import pyarrow.dataset as ds import pytest -from datafusion.object_store import LocalFileSystem, Http +from datafusion.object_store import Http from datafusion import udf, col @@ -462,7 +462,6 @@ def test_register_listing_table( dir_root = f"file://{dir_root}/" if path_to_str else dir_root - ctx.register_object_store("file://local", LocalFileSystem()) ctx.register_listing_table( "my_table", dir_root, diff --git a/python/datafusion/tests/test_store.py b/python/datafusion/tests/test_store.py index d073c613..f85b2831 100644 --- a/python/datafusion/tests/test_store.py +++ b/python/datafusion/tests/test_store.py @@ -16,21 +16,15 @@ # under the License. import os + import pytest from datafusion import SessionContext -from datafusion.object_store import LocalFileSystem - - -@pytest.fixture -def local(): - return LocalFileSystem() @pytest.fixture -def ctx(local): +def ctx(): ctx = SessionContext() - ctx.register_object_store("file://local", local) return ctx