Merge 77f8a45 into 7d1a303

moj-analytical-services · Nov 8, 2021 · bedc977 · bedc977
2 parents 7d1a303 + 77f8a45
commit bedc977
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "splink"
-version = "2.0.0"
+version = "2.0.1"
 description = "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/case_statements.py b/splink/case_statements.py
@@ -1,5 +1,6 @@
 import warnings
 import re
+from .jar_location import similarity_jar_location
 
 
 def _check_jaro_registered(spark):
@@ -32,9 +33,7 @@ def _get_spark_jars_string():
 
     """
 
-    import splink
-
-    path = splink.__file__[0:-11] + "jars/scala-udf-similarity-0.0.9.jar"
+    path = similarity_jar_location()
 
     message = (
         "You will need to add it by correctly configuring your spark config\n"

diff --git a/splink/jar_location.py b/splink/jar_location.py
@@ -0,0 +1,5 @@
+def similarity_jar_location():
+    import splink
+
+    path = splink.__file__[0:-11] + "jars/scala-udf-similarity-0.0.9.jar"
+    return path