kaggle-environments 1.14.8__py2.py3-none-any.whl → 1.14.9__py2.py3-none-any.whl

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/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.dist-info → kaggle_environments-1.14.9.dist-info}/METADATA

@@ -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.dist-info → kaggle_environments-1.14.9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kaggle_environments/__init__.py,sha256=_XNpfNZWcKf_xHo9cNHtbWb4CJqMqG-ybF_RnI4C6HE,1702
+kaggle_environments/__init__.py,sha256=-cMWlmk778J5lbZYSW-yQhB6Ai_NOhDZDiRJd0SL13k,1702
 kaggle_environments/agent.py,sha256=BpYrAC6EOhJF3Wzynq-CWg8l-z5uXW1Lq4qUsyOM0ik,6203
 kaggle_environments/api.py,sha256=eLBKqr11Ku4tdsMUdUqy74FIVEA_hdV3_QUpX84x3Z8,798
 kaggle_environments/core.py,sha256=IrEkN9cIA2djBAxI8Sz1GRpGNKjhqbnBdV6irAeTm8Q,27851
@@ -72,7 +72,7 @@ kaggle_environments/envs/kore_fleets/starter_bots/ts/test/observation.json,sha25
 kaggle_environments/envs/llm_20_questions/keywords.py,sha256=BnyVrV0_xbiDt_Td4YXSKae9hHndAo5YYfYVtyL4eS4,45282
 kaggle_environments/envs/llm_20_questions/llm_20_questions.js,sha256=o30Nx1Tt0eAxBsQukmyxYlr7GuvptTUADWUnEnYDHe8,6009
 kaggle_environments/envs/llm_20_questions/llm_20_questions.json,sha256=VYxgQEDCaJv910sQdE_FOtV2w1V83Y3C6nTjx4FFZyo,2101
-kaggle_environments/envs/llm_20_questions/llm_20_questions.py,sha256=zP8ut87PKgiNeZR6st4Lch5vzLAggMcLbB1kPIB4T5k,9430
+kaggle_environments/envs/llm_20_questions/llm_20_questions.py,sha256=vLNs0yKo86miW2TUYsmWyfOYn9f-jH-H7_krQ3neeHY,9873
 kaggle_environments/envs/lux_ai_2021/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kaggle_environments/envs/lux_ai_2021/agents.py,sha256=bDqsY17Jxrk69a8HXb96ex2U1ipKkQPVJuXxmiA5hMA,421
 kaggle_environments/envs/lux_ai_2021/index.html,sha256=4Ec1FuHuuH1EvkvlLOpNUUFYLovnS_IA-rOOHEUjcGY,1550
@@ -176,9 +176,9 @@ kaggle_environments/envs/tictactoe/tictactoe.js,sha256=NZDT-oSG0a6a-rso9Ldh9qkJw
 kaggle_environments/envs/tictactoe/tictactoe.json,sha256=zMXZ8-fpT7FBhzz2FFBvRLn4XwtngjEqOieMvI6cCj8,1121
 kaggle_environments/envs/tictactoe/tictactoe.py,sha256=iLNU5V-lz7Xab-d1vpPMfU5jDM3QtgBUH63Y_SU7I9Y,3639
 kaggle_environments/static/player.html,sha256=HH8qvFfTIDw8eZvw5W88jcCu58Lo4_eUQ1ak46KWVGQ,22945
-kaggle_environments-1.14.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kaggle_environments-1.14.8.dist-info/METADATA,sha256=GpRi40uIxtp4vv9hJR0DD9FNOlkLH8MbVleNwvduV1s,10700
-kaggle_environments-1.14.8.dist-info/WHEEL,sha256=m9WAupmBd2JGDsXWQGJgMGXIWbQY3F5c2xBJbBhq0nY,110
-kaggle_environments-1.14.8.dist-info/entry_points.txt,sha256=HbVC-LKGQFV6lEEYBYyDTtrkHgdHJUWQ8_qt9KHGqz4,70
-kaggle_environments-1.14.8.dist-info/top_level.txt,sha256=v3MMWIPMQFcI-WuF_dJngHWe9Bb2yH_6p4wat1x4gAc,20
-kaggle_environments-1.14.8.dist-info/RECORD,,
+kaggle_environments-1.14.9.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kaggle_environments-1.14.9.dist-info/METADATA,sha256=9FBsEf3b8jV3jB8F8N6_bQtqPqQYTHQ6hjmfomTgzMo,10700
+kaggle_environments-1.14.9.dist-info/WHEEL,sha256=m9WAupmBd2JGDsXWQGJgMGXIWbQY3F5c2xBJbBhq0nY,110
+kaggle_environments-1.14.9.dist-info/entry_points.txt,sha256=HbVC-LKGQFV6lEEYBYyDTtrkHgdHJUWQ8_qt9KHGqz4,70
+kaggle_environments-1.14.9.dist-info/top_level.txt,sha256=v3MMWIPMQFcI-WuF_dJngHWe9Bb2yH_6p4wat1x4gAc,20
+kaggle_environments-1.14.9.dist-info/RECORD,,