edit

fidelity · Aug 16, 2022 · eeb9f70 · eeb9f70
1 parent 0d36c29
commit eeb9f70
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mab2rec/_version.py b/mab2rec/_version.py
@@ -4,5 +4,5 @@
 
 __author__ = "FMR LLC"
 __email__ = "[email protected]"
-__version__ = "1.0.2"
+__version__ = "1.0.3"
 __copyright__ = "Copyright (C), FMR LLC"
diff --git a/mab2rec/pipeline.py b/mab2rec/pipeline.py
@@ -238,13 +238,14 @@ def score(recommender: Union[str, BanditRecommender],
     _validate_save(save_file)
 
     # Import data
+    item_list_out = recommender.mab.arms if item_list is None else item_list
     test_data_df, item_list_out, user_features_df, \
         item_to_features, excluded_df = load_data(data=data,
                                                   user_features=user_features,
                                                   user_features_list=user_features_list,
                                                   user_features_dtypes=user_features_dtypes,
                                                   item_features=item_features,
-                                                  item_list=item_list,
+                                                  item_list=item_list_out,
                                                   item_eligibility=item_eligibility,
                                                   user_id_col=user_id_col,
                                                   item_id_col=item_id_col,