bump version to 0.4.10 (#757)

This commit is contained in:
Jiayi Weng 2022-10-16 22:15:20 -07:00 committed by GitHub
parent 0181fe79a5
commit 41ae3461f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View File

@ -4,6 +4,7 @@ from copy import deepcopy
from typing import Optional, Tuple from typing import Optional, Tuple
import gym import gym
import gymnasium
import numpy as np import numpy as np
import torch import torch
from pettingzoo.classic import tictactoe_v3 from pettingzoo.classic import tictactoe_v3
@ -104,7 +105,7 @@ def get_agents(
) -> Tuple[BasePolicy, torch.optim.Optimizer, list]: ) -> Tuple[BasePolicy, torch.optim.Optimizer, list]:
env = get_env() env = get_env()
observation_space = env.observation_space['observation'] if isinstance( 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 ) else env.observation_space
args.state_shape = observation_space.shape or observation_space.n args.state_shape = observation_space.shape or observation_space.n
args.action_shape = env.action_space.shape or env.action_space.n args.action_shape = env.action_space.shape or env.action_space.n

View File

@ -1,6 +1,6 @@
from tianshou import data, env, exploration, policy, trainer, utils from tianshou import data, env, exploration, policy, trainer, utils
__version__ = "0.4.9" __version__ = "0.4.10"
__all__ = [ __all__ = [
"env", "env",