Follow gym 0.25 spec

This commit is contained in:
Dominik Moritz Roth 2022-12-13 20:13:53 +01:00
parent 048ba027f3
commit d8e7c4c80f

View File

@ -13,7 +13,8 @@ from columbus.utils import soft_int, parseObs
class ColumbusEnv(gym.Env):
metadata = {'render.modes': ['human']}
metadata = {'render.modes': ['human'], 'render_modes': [
'human', 'non-human'], 'render_fps': 60}
def __init__(self, observable=observables.Observable(), fps=60, env_seed=3.1, master_seed=None, start_pos=(0.5, 0.5), start_score=0, speed_fac=0.01, acc_fac=0.04, die_on_zero=False, return_on_score=-1, reward_mult=1, agent_drag=0, controll_type='SPEED', aux_reward_max=1, aux_penalty_max=0, aux_reward_discretize=0, void_is_type_barrier=True, void_damage=1, torus_topology=False, default_collision_elasticity=1, terminate_on_reward=False, agent_draw_path=False, clear_path_on_reset=True, max_steps=-1, value_color_mapper='tanh', width=720, height=720, agent_attrs={}, agent_cls=entities.Agent, exception_for_unsupported_collision=True):
super(ColumbusEnv, self).__init__()