diff --git a/aicrowd.json b/aicrowd.json new file mode 100644 index 0000000000000000000000000000000000000000..23a1aec3631d8965353c2dabd03e66b76e35c76b --- /dev/null +++ b/aicrowd.json @@ -0,0 +1,8 @@ +{ + "challenge_id": "user-behavior-alignment", + "authors": [ + "aicrowd-bot" + ], + "gpu": true, + "description": "(optional) description about your awesome agent" +} \ No newline at end of file diff --git a/models/dummy_model.py b/models/dummy_model.py index b18dd24cbaf83a9b7c2227e43b2cdc5fd91665d7..2eb84d3e52d0deb7d8f633090b5a703e79636446 100644 --- a/models/dummy_model.py +++ b/models/dummy_model.py @@ -19,7 +19,7 @@ class DummyModel: """ return 0 - def task_ranking(self, task_prompt: str) -> List[float]: + def task_ranking(self, task_prompt: str) -> List[int]: """ Task method for Ranking Input - Task Prompt (includes items to rank)