From 5c4303deece56d3cc37cfcccecb40ce169df0d52 Mon Sep 17 00:00:00 2001 From: Richard Tia Date: Fri, 22 Mar 2024 10:07:24 -0700 Subject: [PATCH] fix: fix imports for tpch tests --- substrait_consumer/tests/integration/test_acero_tpch.py | 3 ++- substrait_consumer/tests/integration/test_duckdb_tpch.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/substrait_consumer/tests/integration/test_acero_tpch.py b/substrait_consumer/tests/integration/test_acero_tpch.py index deae3f75..04529c27 100644 --- a/substrait_consumer/tests/integration/test_acero_tpch.py +++ b/substrait_consumer/tests/integration/test_acero_tpch.py @@ -6,7 +6,8 @@ from pyarrow import compute from substrait_consumer.common import SubstraitUtils -from substrait_consumer.consumers import AceroConsumer, DuckDBConsumer +from substrait_consumer.consumers.acero_consumer import AceroConsumer +from substrait_consumer.consumers.duckdb_consumer import DuckDBConsumer from substrait_consumer.parametrization import custom_parametrization from substrait_consumer.verification import verify_equals from substrait_consumer.tests.integration.queries.tpch_test_cases import TPCH_QUERY_TESTS diff --git a/substrait_consumer/tests/integration/test_duckdb_tpch.py b/substrait_consumer/tests/integration/test_duckdb_tpch.py index e43af15b..a83aebab 100644 --- a/substrait_consumer/tests/integration/test_duckdb_tpch.py +++ b/substrait_consumer/tests/integration/test_duckdb_tpch.py @@ -1,7 +1,7 @@ import duckdb import pytest -from substrait_consumer.consumers import DuckDBConsumer +from substrait_consumer.consumers.duckdb_consumer import DuckDBConsumer from substrait_consumer.parametrization import custom_parametrization from substrait_consumer.verification import verify_equals from .queries.tpch_test_cases import TPCH_QUERY_TESTS