diff --git a/pyproject.toml b/pyproject.toml index b7a51e0e80..bb8c7bf26b 100644 --- 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 ", "Sam Lindsay", "Theodore Manassis"] license = "MIT" diff --git a/splink/intuition.py b/splink/intuition.py index a3d0082c6b..d4b65211ad 100644 --- 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 index 2512d928e5..8f9a16f06f 100644 --- 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() \ No newline at end of file + 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)) +