From: Nathan TeBlunthuis <nathante@uw.edu>
Date: Wed, 31 May 2023 16:47:21 +0000 (-0700)
Subject: make pass keyword arg to dataframe.drop
X-Git-Url: https://code.communitydata.science/cdsc_reddit.git/commitdiff_plain/refs/heads/icwsm_dataverse

make pass keyword arg to dataframe.drop
---

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)