bump version to 0.4.10 (#757)
This commit is contained in:
parent
0181fe79a5
commit
41ae3461f6
@ -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
|
||||||
|
@ -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",
|
||||||
|
Loading…
x
Reference in New Issue
Block a user