fix input (#13)

Signed-off-by: Kleynhans, Bernard <[email protected]>
fidelity · Jul 5, 2022 · d22227d · d22227d
1 parent 95a6f3f
commit d22227d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/CHANGELOG.txt b/CHANGELOG.txt
@@ -2,6 +2,13 @@
 CHANGELOG
 =========
 
+-------------------------------------------------------------------------------
+July, 5, 2022 1.0.2
+-------------------------------------------------------------------------------
+
+Minor:
+- Fixed bug with arguments not being passed correctly to load_response_data.
+
 -------------------------------------------------------------------------------
 May, 12, 2022 1.0.1
 -------------------------------------------------------------------------------

diff --git a/mab2rec/_version.py b/mab2rec/_version.py
@@ -4,5 +4,5 @@
 
 __author__ = "FMR LLC"
 __email__ = "[email protected]"
-__version__ = "1.0.1"
+__version__ = "1.0.2"
 __copyright__ = "Copyright (C), FMR LLC"
diff --git a/mab2rec/pipeline.py b/mab2rec/pipeline.py
@@ -481,7 +481,7 @@ def _bench(recommenders: Dict[str, BanditRecommender],
                                         user_id_col=user_id_col,
                                         item_id_col=item_id_col,
                                         response_col=response_col)
-    test_data_df = load_response_data(test_data)
+    test_data_df = load_response_data(test_data, user_id_col, item_id_col, response_col)
 
     recommendations = dict()
     rec_metrics = dict()