CR

Kaggle · Jun 26, 2024 · dc0c8af · dc0c8af
1 parent a232bd1
commit dc0c8af
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/kaggle_environments/envs/llm_20_questions/llm_20_questions.py b/kaggle_environments/envs/llm_20_questions/llm_20_questions.py
@@ -36,10 +36,9 @@
 RANDOM_ANSWERER = "random_answerer"
 
 def weighted_random_category(keywords_list):
-    try:   
-        return random.choice(keywords_list, weights=[len(entry["words"]) for entry in keywords_list], k=1)[0]
+    try:
+        return random.choices(population=keywords_list, weights=[len(entry["words"]) for entry in keywords_list], k=1)[0]
     except:
-        print("FAIL")
         pass
     return random.choice(keywords_list)