fix failed concatenation in newer numpy ref #10

david-cortes · May 7, 2021 · e807f61 · e807f61
1 parent 13a9077
commit e807f61
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hpfrec/__init__.py b/hpfrec/__init__.py
@@ -1156,7 +1156,7 @@ def add_user(self, user_id, counts_df, update_existing=False, maxiter=10, ncores
 			else:
 				if self.reindex:
 					new_id = self.user_mapping_.shape[0]
-					self.user_mapping_ = np.r_[self.user_mapping_, user_id]
+					self.user_mapping_ = np.r_[self.user_mapping_, np.array(user_id)]
 					if self.produce_dicts:
 						self.user_dict_[user_id] = new_id
 				self.Theta = np.r_[self.Theta, Theta.reshape((1, self.k))]
@@ -1175,7 +1175,7 @@ def add_user(self, user_id, counts_df, update_existing=False, maxiter=10, ncores
 				self.seen = np.r_[self.seen[:user_id], counts_df.ItemId.values, self.seen[(user_id + 1):]]
 				self._st_ix_user[(user_id + 1):] += self._n_seen_by_user[user_id] - n_seen_by_user_before
 			else:
-				self._n_seen_by_user = np.r_[self._n_seen_by_user, counts_df.shape[0]]
+				self._n_seen_by_user = np.r_[self._n_seen_by_user, np.array(counts_df.shape[0])]
 				self._st_ix_user = np.r_[self._st_ix_user, self.seen.shape[0]]
 				self.seen = np.r_[self.seen, counts_df.ItemId.values]
 

diff --git a/setup.py b/setup.py
@@ -58,7 +58,7 @@ def build_extensions(self):
 	 'scipy',
 	 'cython'
 ],
-	version = '0.2.3.3',
+	version = '0.2.3.4',
 	description = 'Hierarchical Poisson matrix factorization for recommender systems',
 	author = 'David Cortes',
 	author_email = 'david.cortes.rivera@gmail.com',