From b947a52d9cdd31536320a3706072d704e02e7521 Mon Sep 17 00:00:00 2001 From: Robin Linacre Date: Mon, 25 Mar 2024 20:35:28 +0000 Subject: [PATCH] fix typo --- 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 225d21512d..826bc20dae 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(reusable=False) pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.execute_pipeline(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(reusable=False) pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.execute_pipeline(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(reusable=False) pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.execute_pipeline(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(reusable=False) pipeline.enqueue_sql(check_blocking_sql, "__splink__df_blocked_same_table_count") - self_table_count = linker.db_api.execute_pipeline(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()