Skip to content

Commit 7aa9bcf

Browse files
authored
NO-SNOW: fix test failure (#3903)
1 parent 85466f9 commit 7aa9bcf

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

src/snowflake/snowpark/_internal/data_source/drivers/base_driver.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,7 @@ def udtf_ingestion(
177177
packages=packages or UDTF_PACKAGE_MAP.get(self.dbms_type),
178178
imports=imports,
179179
statement_params=statement_params,
180+
_emit_ast=_emit_ast, # internal function call, _emit_ast will be set to False by the caller
180181
)
181182
logger.debug(f"register ingestion udtf takes: {udtf_register_time()} seconds")
182183
call_udtf_sql = f"""

tests/integ/test_data_source_api.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@
7979
sql_server_create_connection_unicode_data,
8080
sql_server_create_connection_double_quoted_data,
8181
)
82-
from tests.utils import Utils, IS_WINDOWS
82+
from tests.utils import Utils, IS_WINDOWS, RUNNING_ON_JENKINS
8383

8484
try:
8585
import pandas # noqa: F401
@@ -496,8 +496,14 @@ def assert_datasource_statement_params_run_query(*args, **kwargs):
496496
assert called == 2
497497

498498

499-
def test_telemetry_tracking_for_udtf(caplog, session):
499+
@pytest.mark.skipif(
500+
RUNNING_ON_JENKINS,
501+
reason="SNOW-2089683: oracledb real connection test failed on jenkins",
502+
)
503+
def test_telemetry_tracking_for_udtf(caplog, session, ast_enabled):
500504

505+
if ast_enabled:
506+
pytest.skip("TODO: dbapi has not implemented ast yet, skip the test for now")
501507
original_func = session._conn.run_query
502508
called = 0
503509

0 commit comments

Comments
 (0)