fix typo

moj-analytical-services · Mar 25, 2024 · b947a52 · b947a52
1 parent 8c2d2d4
commit b947a52
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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()