intuiition report bug

moj-analytical-services · Mar 17, 2020 · 826c4be · 826c4be
1 parent afb7c60
commit 826c4be
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "splink"
-version = "0.1.5"
+version = "0.1.6"
 description = "[Beta]: Implementation in Apache Spark of the EM algorithm to estimate parameters of Fellegi-Sunter's canonical model of record linkage."
 authors = ["Robin Linacre <robinlinacre@hotmail.com>", "Sam Lindsay", "Theodore Manassis"]
 license = "MIT"

diff --git a/splink/intuition.py b/splink/intuition.py
@@ -1,5 +1,5 @@
 from .chart_definitions import adjustment_factor_chart_def
-from params import Params
+from .params import Params
 
 altair_installed = True
 try:

diff --git a/tests/test_spark.py b/tests/test_spark.py
@@ -591,4 +591,10 @@ def test_main_api(spark, sqlite_con_1):
     df_e = linker.get_scored_comparisons()
     linker.save_model_as_json("saved_model.json", overwrite=True)
     linker_2 = load_from_json("saved_model.json", spark=spark, df=df)
-    df_e = linker_2.get_scored_comparisons()
+    df_e = linker_2.get_scored_comparisons()
+
+    from splink.intuition import intuition_report
+    params = linker.params
+    row_dict = df_e.toPandas().sample(1).to_dict(orient="records")[0]
+    print(intuition_report(row_dict, params))
+