Merge pull request #41 from andlzanon/separator

Separator bug fix to read all datasets
caserec · Jun 17, 2021 · f7f4369 · f7f4369
2 parents d39157d + 1a0a329
commit f7f4369
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caserec/recommenders/item_recommendation/base_item_recommendation.py b/caserec/recommenders/item_recommendation/base_item_recommendation.py
@@ -83,7 +83,7 @@ def read_files(self):
         self.train_set = ReadFile(self.train_file, sep=self.sep, as_binary=self.as_binary).read()
 
         if self.test_file is not None:
-            self.test_set = ReadFile(self.test_file).read()
+            self.test_set = ReadFile(self.test_file, sep=self.sep).read()
             self.users = sorted(set(list(self.train_set['users']) + list(self.test_set['users'])))
             self.items = sorted(set(list(self.train_set['items']) + list(self.test_set['items'])))
         else: