diff --git a/test/pettingzoo/tic_tac_toe.py b/test/pettingzoo/tic_tac_toe.py index 21d44ffa2..62d9a0905 100644 --- a/test/pettingzoo/tic_tac_toe.py +++ b/test/pettingzoo/tic_tac_toe.py @@ -4,6 +4,7 @@ from typing import Optional, Tuple import gym +import gymnasium import numpy as np import torch from pettingzoo.classic import tictactoe_v3 @@ -104,7 +105,7 @@ def get_agents( ) -> Tuple[BasePolicy, torch.optim.Optimizer, list]: env = get_env() observation_space = env.observation_space['observation'] if isinstance( - env.observation_space, gym.spaces.Dict + env.observation_space, (gym.spaces.Dict, gymnasium.spaces.Dict) ) else env.observation_space args.state_shape = observation_space.shape or observation_space.n args.action_shape = env.action_space.shape or env.action_space.n diff --git a/tianshou/__init__.py b/tianshou/__init__.py index 281de6f8a..f0cc7c75d 100644 --- a/tianshou/__init__.py +++ b/tianshou/__init__.py @@ -1,6 +1,6 @@ from tianshou import data, env, exploration, policy, trainer, utils -__version__ = "0.4.9" +__version__ = "0.4.10" __all__ = [ "env",