make sure strings are suitable length before checking for plurals (#278)

Fixes https://www.kaggle.com/competitions/llm-20-questions/discussion/513081
Kaggle · Jun 18, 2024 · ecdf7a3 · ecdf7a3
1 parent c6ccfb8
commit ecdf7a3
Showing 1 changed file with 3 additions and 0 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
@@ -283,6 +283,9 @@ def compare_words(a, b) -> bool:
     b = normalize(b)
     if a == b:
         return True
+    # don't check for plurals if string is too short
+    if len(a) < 3 or len(b) < 3:
+        return False
     # accept common plurals
     if a[-1] == "s" and a[:-1] == b:
         return True