kaggle-environments 1.14.8__tar.gz → 1.14.9__tar.gz

{kaggle-environments-1.14.8/kaggle_environments.egg-info → kaggle-environments-1.14.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kaggle-environments
-Version: 1.14.8
+Version: 1.14.9
 Summary: Kaggle Environments
 Home-page: https://github.com/Kaggle/kaggle-environments
 Author: Kaggle

{kaggle-environments-1.14.8 → kaggle-environments-1.14.9}/kaggle_environments/__init__.py

@@ -21,7 +21,7 @@ from .main import http_request
 from . import errors
 from . import utils
 
-__version__ = "1.14.8"
+__version__ = "1.14.9"
 
 __all__ = ["Agent", "environments", "errors", "evaluate", "http_request",
            "make", "register", "utils", "__version__",

{kaggle-environments-1.14.8 → kaggle-environments-1.14.9}/kaggle_environments/envs/llm_20_questions/llm_20_questions.py

@@ -8,7 +8,7 @@ import torch
 from .keywords import KEYWORDS_JSON
 from os import path
 from pathlib import Path
-from random import choice
+from random import choice, random
 from string import Template
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
@@ -29,7 +29,9 @@ TIMEOUT = "TIMEOUT"
 GUESS = "guess"
 ASK = "ask"
 GUESSER = "guesser"
+RANDOM_GUESSER = "random_guesser"
 ANSWERER = "answerer"
+RANDOM_ANSWERER = "random_answerer"
 
 keywords_list = json.loads(KEYWORDS_JSON)
 keyword_cat = random.choice(keywords_list)
@@ -68,6 +70,19 @@ def guesser_agent(obs):
     
     return call_llm(prompt)
 
+def random_guesser(obs):
+    if obs.turnType == GUESS:
+        return "banana"
+    if random() < .5:
+        return "Is is a person?"
+    if random() < .5:
+        return "Is it a place?"
+    return "Is it a thing?"
+
+def random_answerer():
+    if random() > .5:
+        return "yes"
+    return "no"
     
 
 def answerer_agent(obs):
@@ -84,7 +99,7 @@ def answerer_agent(obs):
         return ""
 
 
-agents = {GUESSER: guesser_agent, ANSWERER: answerer_agent}
+agents = {GUESSER: guesser_agent, ANSWERER: answerer_agent, RANDOM_ANSWERER: random_answerer, RANDOM_GUESSER: random_guesser}
 
 def guesser_action(active, inactive, step):
     inactive.observation.keyword = keyword

{kaggle-environments-1.14.8 → kaggle-environments-1.14.9/kaggle_environments.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kaggle-environments
-Version: 1.14.8
+Version: 1.14.9
 Summary: Kaggle Environments
 Home-page: https://github.com/Kaggle/kaggle-environments
 Author: Kaggle