From 573358f5e6a611102d8490917b9f66f973b9a171 Mon Sep 17 00:00:00 2001 From: Phillip Cloud <417981+cpcloud@users.noreply.github.com> Date: Mon, 27 Nov 2023 07:46:16 -0500 Subject: [PATCH] chore: fix sql tests import --- ibis/tests/sql/conftest.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/ibis/tests/sql/conftest.py b/ibis/tests/sql/conftest.py index 10eb4161f2ab..7f0087081b79 100644 --- a/ibis/tests/sql/conftest.py +++ b/ibis/tests/sql/conftest.py @@ -4,9 +4,7 @@ import sqlglot as sg import ibis - -# from ibis.backends.base.sql.compiler import Compiler, QueryContext -from ibis.backends.duckdb.compiler import translate +from ibis.backends.duckdb import Backend as DuckDBBackend from ibis.tests.expr.mocks import MockBackend @@ -78,7 +76,7 @@ def to_sql(expr, *args, **kwargs) -> str: expr = expr.optimize() - sql = translate(expr.op(), params={}) + sql = DuckDBBackend.compiler.translate(expr.op(), params={}) if isinstance(sql, sg.exp.Table): sql = sg.select("*").from_(sql)