From 27d2a7eb6d990dd6fddc223ffa3be7c968adf5e5 Mon Sep 17 00:00:00 2001 From: Argenis Leon Date: Tue, 26 Nov 2019 08:19:32 -0600 Subject: [PATCH] Fixed args --- optimus/ml/keycollision.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/optimus/ml/keycollision.py b/optimus/ml/keycollision.py index a1d9c57f..e2830abd 100644 --- a/optimus/ml/keycollision.py +++ b/optimus/ml/keycollision.py @@ -89,15 +89,14 @@ def calculate_ngrams(value, args): def fingerprint_cluster(df, input_cols, output: str = "dict"): - return base_cluster(df, input_cols, output, func=fingerprint, args=[input_cols]) + return base_clustering_function(df, input_cols, output, func=fingerprint, args=[input_cols]) def n_gram_fingerprint_cluster(df, input_cols, n_size=2, output: str = "dict"): - return base_cluster(df, input_cols, output, func=n_gram_fingerprint, args=[input_cols, n_size]) + return base_clustering_function(df, input_cols, output, func=n_gram_fingerprint, args=[input_cols, n_size]) -# df, input_cols, output: str = "dict" -def base_cluster(df, input_cols, output, func=None, args=None): +def base_clustering_function(df, input_cols, output, func=None, args=None): """ Cluster a dataframe column based on the Fingerprint algorithm :return: