From 9345f9de9437d5965ad4ee5874bc24199e077d48 Mon Sep 17 00:00:00 2001 From: Nathan TeBlunthuis Date: Wed, 31 May 2023 09:47:21 -0700 Subject: [PATCH] make pass keyword arg to dataframe.drop --- clustering/clustering_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clustering/clustering_base.py b/clustering/clustering_base.py index 98a260e..2f37b68 100644 --- a/clustering/clustering_base.py +++ b/clustering/clustering_base.py @@ -68,7 +68,7 @@ class clustering_job: def read_distance_mat(self, similarities, use_threads=True): print(similarities) df = pd.read_feather(similarities, use_threads=use_threads) - mat = np.array(df.drop('_subreddit',1)) + mat = np.array(df.drop('_subreddit',axis=1)) n = mat.shape[0] mat[range(n),range(n)] = 1 return (df._subreddit,1-mat)