fix more tests

moj-analytical-services · Apr 23, 2024 · 2e237de · 2e237de
1 parent 54e6597
commit 2e237de
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/docs/charts/accuracy_chart_from_labels_table.ipynb b/docs/charts/accuracy_chart_from_labels_table.ipynb
@@ -147,7 +147,7 @@
     "df_labels = splink_dataset_labels.fake_1000_labels\n",
     "labels_table = linker.register_labels_table(df_labels)\n",
     "\n",
-    "linker.accuracy_chart_from_labels_table(labels_table, add_metrics=['f1'])"
+    "linker.accuracy_analysis_from_labels_table(labels_table, output_type=\"accuracy\", add_metrics=['f1'])"
    ]
   },
   {

diff --git a/docs/charts/threshold_selection_tool_from_labels_table.ipynb b/docs/charts/threshold_selection_tool_from_labels_table.ipynb
@@ -147,7 +147,7 @@
     "df_labels = splink_dataset_labels.fake_1000_labels\n",
     "labels_table = linker.register_labels_table(df_labels)\n",
     "\n",
-    "linker.threshold_selection_tool_from_labels_table(labels_table, add_metrics=['f1'])"
+    "linker.accuracy_analysis_from_labels_table(labels_table, add_metrics=['f1'])"
    ]
   },
   {

diff --git a/tests/test_full_example_duckdb.py b/tests/test_full_example_duckdb.py
@@ -90,7 +90,7 @@ def test_full_example_duckdb(tmp_path):
 
     register_roc_data(linker)
 
-    linker.threshold_selection_tool_from_labels_table("labels")
+    linker.accuracy_analysis_from_labels_table("labels")
 
     df_clusters = linker.cluster_pairwise_predictions_at_threshold(df_predict, 0.1)
 

diff --git a/tests/test_full_example_postgres.py b/tests/test_full_example_postgres.py
@@ -70,7 +70,7 @@ def test_full_example_postgres(tmp_path, pg_engine):
 
     register_roc_data(linker)
 
-    linker.threshold_selection_tool_from_labels_table("labels")
+    linker.accuracy_analysis_from_labels_table("labels")
 
     df_clusters = linker.cluster_pairwise_predictions_at_threshold(df_predict, 0.1)
 

diff --git a/tests/test_full_example_spark.py b/tests/test_full_example_spark.py
@@ -122,7 +122,7 @@ def spark_csv_read(x):
     )
     register_roc_data(linker)
 
-    linker.threshold_selection_tool_from_labels_table("labels")
+    linker.accuracy_analysis_from_labels_table("labels")
 
     record = {
         "unique_id": 1,

diff --git a/tests/test_full_example_sqlite.py b/tests/test_full_example_sqlite.py
@@ -60,7 +60,7 @@ def test_full_example_sqlite(tmp_path):
 
     register_roc_data(linker)
 
-    linker.threshold_selection_tool_from_labels_table("labels")
+    linker.accuracy_analysis_from_labels_table("labels")
 
 
 @mark_with_dialects_including("sqlite")