Update crs_arena/crs_fighter.py

crs_arena/crs_fighter.py

@@ -45,14 +45,10 @@ class CRSFighter:
 
         # Generation arguments
         self.response_generation_args = {}
-        if self.name.split("_")[0] == "unicrs":
+        if self.name.split("_")[0].__contains__("unicrs"):
             self.response_generation_args.update(
                 {
-                    "movie_token": (
-                        "<movie>"
-                        if self.model.crs_model.kg_dataset.startswith("redial")
-                        else "<mask>"
-                    ),
+                    "movie_token": "<pad>",
                 }
             )