From be86488fde9d19163d88197694af79a4f1515a9c Mon Sep 17 00:00:00 2001 From: Robin Linacre Date: Fri, 7 Jun 2024 10:35:01 +0100 Subject: [PATCH] fix tests --- tests/test_u_train.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_u_train.py b/tests/test_u_train.py index b4bfb9f3b4..89d3ce69b6 100644 --- a/tests/test_u_train.py +++ b/tests/test_u_train.py @@ -90,7 +90,7 @@ def test_u_train_link_only(test_helpers, dialect): pipeline = CTEPipeline() pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.sql_pipeline_to_splink_dataframe(pipeline) + self_table_count = linker._db_api.sql_pipeline_to_splink_dataframe(pipeline) result = self_table_count.as_record_dict() self_table_count.drop_table_from_database_and_remove_from_cache() @@ -152,7 +152,7 @@ def test_u_train_link_only_sample(test_helpers, dialect): pipeline = CTEPipeline() pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.sql_pipeline_to_splink_dataframe(pipeline) + self_table_count = linker._db_api.sql_pipeline_to_splink_dataframe(pipeline) result = self_table_count.as_record_dict() self_table_count.drop_table_from_database_and_remove_from_cache() @@ -277,7 +277,7 @@ def test_u_train_multilink(test_helpers, dialect): pipeline = CTEPipeline() pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.sql_pipeline_to_splink_dataframe(pipeline) + self_table_count = linker._db_api.sql_pipeline_to_splink_dataframe(pipeline) result = self_table_count.as_record_dict() self_table_count.drop_table_from_database_and_remove_from_cache() @@ -309,7 +309,7 @@ def test_u_train_multilink(test_helpers, dialect): pipeline = CTEPipeline() pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.sql_pipeline_to_splink_dataframe(pipeline) + self_table_count = linker._db_api.sql_pipeline_to_splink_dataframe(pipeline) result = self_table_count.as_record_dict() self_table_count.drop_table_from_database_and_remove_from_cache()