Upload main.py

main.py

@@ -78,7 +78,7 @@ def retrieve_candidates(query: str, n_candidates: int):
     return candidates_idx
 
 
-def rerank(candidates_idx: np.ndarray, n_rec: int = 10, data_source: str) -> list:
+def rerank(candidates_idx: np.ndarray, n_rec: int = 10, data_source: str = None) -> list:
     
     # Get popularity scores for stage 1 candidates 
     rerank_scores = data.loc[candidates_idx, "pop_score"].values