Spaces:

Artvv
/

PersistentPoker-Bench

Sleeping

PersistentPoker-Bench / src /persistentpoker_bench /testsupport.py

Upload src/persistentpoker_bench/testsupport.py with huggingface_hub

53cd6b0 verified 6 months ago

585 Bytes

from __future__ import annotations

from persistentpoker_bench.hand_runner import StaticDecisionAgent
from persistentpoker_bench.schemas import LLMDecision, WinnerPoolDecision


def static_agent_factory() -> StaticDecisionAgent:
    # Keep demo/smoke agents deterministic but with enough scripted actions to
    # survive longer multi-hand runs without exhausting the stub.
    decisions = [LLMDecision("call", None, (), WinnerPoolDecision.CONTINUE)] * 1024 + [
        LLMDecision("check", None, (), WinnerPoolDecision.CONTINUE)
    ] * 2048
    return StaticDecisionAgent(decisions)

	from __future__ import annotations

	from persistentpoker_bench.hand_runner import StaticDecisionAgent
	from persistentpoker_bench.schemas import LLMDecision, WinnerPoolDecision


	def static_agent_factory() -> StaticDecisionAgent:
	# Keep demo/smoke agents deterministic but with enough scripted actions to
	# survive longer multi-hand runs without exhausting the stub.
	decisions = [LLMDecision("call", None, (), WinnerPoolDecision.CONTINUE)] * 1024 + [
	LLMDecision("check", None, (), WinnerPoolDecision.CONTINUE)
	] * 2048
	return StaticDecisionAgent(decisions)