Update recommender/recommender.py

recommender/recommender.py

@@ -118,10 +118,9 @@ class Recommender:
         df_final = df_final.sort_values('final_score', ascending=False)
 
         # Itinerary planning
-        #must_haves = extract_must_haves(sub_queries, df_final)
-        #if must_haves:
-            #print(f"Applying hard constraints: {must_haves}")
-            #return self.planner.plan_day(df_final, mode="or_tools", must_haves=must_haves)
-        #else:
-        print(" [TEST MODE] Forcing Beam Search Heuristic...")
-        return self.planner.plan_day(df_final, mode="beam", beam_width=3)
+        must_haves = extract_must_haves(sub_queries, df_final)
+        if must_haves:
+            print(f"Applying hard constraints: {must_haves}")
+            return self.planner.plan_day(df_final, mode="or_tools", must_haves=must_haves)
+        else:
+            return self.planner.plan_day(df_final, mode="beam", beam_width=3)