fix mypy errors

fschlatt · Nov 16, 2021 · 2e6cfd1 · 2e6cfd1
1 parent 64dc51a
commit 2e6cfd1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/test/test_gym/test_agent.py b/test/test_gym/test_agent.py
@@ -6,14 +6,14 @@
 import clubs_gym
 
 
-def test_base():
+def test_base() -> None:
     agent = clubs_gym.agent.BaseAgent()
 
     with pytest.raises(NotImplementedError):
         agent.act({})
 
 
-def test_agent():
+def test_agent() -> None:
     agent = clubs_gym.agent.kuhn.NashKuhnAgent(0)
 
     obs = {

diff --git a/test/test_gym/test_env.py b/test/test_gym/test_env.py
@@ -9,7 +9,7 @@
 from clubs_gym import error
 
 
-def test_env():
+def test_env() -> None:
     env = gym.make("NoLimitHoldemTwoPlayer-v0")
     dealer = clubs.poker.Dealer(**clubs.configs.NO_LIMIT_HOLDEM_TWO_PLAYER)
 
@@ -36,7 +36,7 @@ def test_env():
     assert env.close() is None
 
 
-def test_register():
+def test_register() -> None:
     env = gym.make("KuhnTwoPlayer-v0")
 
     with pytest.raises(error.NoRegisteredAgentsError):
@@ -61,7 +61,7 @@ def test_register():
     _ = env.step(action)
 
 
-def test_errors():
+def test_errors() -> None:
     env = gym.make("NoLimitHoldemTwoPlayer-v0")
     with pytest.raises(error.InvalidAgentConfigurationError):
         env.register_agents(None)