From d3953b0f5fce7b4c13ef4acf777621444c2ab8eb Mon Sep 17 00:00:00 2001 From: Daniel Mesejo Date: Fri, 27 Sep 2024 22:38:17 +0200 Subject: [PATCH] chore: remove redundant ObjectStore --- python/datafusion/object_store.py | 4 ---- python/datafusion/tests/test_sql.py | 3 +-- python/datafusion/tests/test_store.py | 10 ++-------- 3 files changed, 3 insertions(+), 14 deletions(-) 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