small bug with saving with reindex=False

david-cortes · May 31, 2018 · ced3893 · ced3893
1 parent 9116ac1
commit ced3893
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hpfrec/__init__.py b/hpfrec/__init__.py
@@ -296,7 +296,7 @@ def fit(self, counts_df, val_set=None):
         ## after terminating optimization
         if self.keep_data:
             self._store_metadata()
-        if self.produce_dicts:
+        if self.produce_dicts and self.reindex:
             self.user_dict_ = {self.user_mapping_[i]:i for i in range(self.user_mapping_.shape[0])}
             self.item_dict_ = {self.item_mapping_[i]:i for i in range(self.item_mapping_.shape[0])}
         self.is_fitted = True