ensure agents are added before reset

dickreuter · Feb 16, 2024 · e54add4 · e54add4
1 parent b493b13
commit e54add4
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 2 deletions.
diff --git a/gym_env/env.py b/gym_env/env.py
@@ -168,6 +168,11 @@ def reset(self):
         self.funds_history = pd.DataFrame()
         self.first_action_for_hand = [True] * len(self.players)
 
+        if not self.players:
+            log.warning("No agents added. Add agents before resetting the environment.")
+            return
+
+
         for player in self.players:
             player.stack = self.initial_stacks
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,6 +17,7 @@ cppimport = "^22.8.2"
 docopt = "^0.6.2"
 pydocstyle = "^6.3.0"
 matplotlib = "^3.8.2"
+mpld3 = "^0.5.10"
 
 
 [build-system]

diff --git a/readme.rst b/readme.rst
@@ -14,7 +14,7 @@ Run:
 - Install Poetry with ``curl -sSL https://install.python-poetry.org | python3 -``
 - Create a virtual environment with ``poetry env use python3.11``
 - Activate it with ``poetry shell``
-- Install all required packages with ``poetry install``
+- Install all required packages with ``poetry install --no-root``
 - Run 6 random players playing against each other:
   ``poetry run python main.py selfplay random --render`` or
 - To manually control the players:``poetry run python main.py selfplay keypress --render``