|
13 | 13 | function GymEnv(id::String) |
14 | 14 | gymenv = nothing |
15 | 15 | try |
16 | | - gymenv = gym[:make](id) |
| 16 | + gymenv = gym.make(id) |
17 | 17 | catch e |
18 | 18 | error("Error received during the initialization of $id\n$e") |
19 | 19 | end |
20 | 20 |
|
21 | | - spec = Spec(gymenv[:spec][:id], |
22 | | - gymenv[:spec][:trials], |
23 | | - gymenv[:spec][:reward_threshold], |
24 | | - gymenv[:spec][:nondeterministic], |
25 | | - gymenv[:spec][:tags], |
26 | | - gymenv[:spec][:max_episode_steps], |
27 | | - gymenv[:spec][:timestep_limit] |
| 21 | + spec = Spec(gymenv.spec.id, |
| 22 | + gymenv.spec.trials, |
| 23 | + gymenv.spec.reward_threshold, |
| 24 | + gymenv.spec.nondeterministic, |
| 25 | + gymenv.spec.tags, |
| 26 | + gymenv.spec.max_episode_steps, |
| 27 | + gymenv.spec.timestep_limit |
28 | 28 | ) |
29 | | - action_space = julia_space(gymenv[:action_space]) |
30 | | - observation_space = julia_space(gymenv[:observation_space]) |
| 29 | + action_space = julia_space(gymenv.action_space) |
| 30 | + observation_space = julia_space(gymenv.observation_space) |
31 | 31 |
|
32 | 32 | env = GymEnv(id, spec, action_space, |
33 | | - observation_space, gymenv[:reward_range], gymenv) |
| 33 | + observation_space, gymenv.reward_range, gymenv) |
34 | 34 | return env |
35 | 35 | end |
36 | 36 |
|
37 | | -reset!(env::GymEnv) = env.gymenv[:reset]() |
| 37 | +reset!(env::GymEnv) = env.gymenv.reset() |
38 | 38 | function render(env::GymEnv; mode="human") |
39 | | - env.gymenv[:render](mode) |
| 39 | + env.gymenv.render(mode) |
40 | 40 | end |
41 | 41 |
|
42 | 42 | function step!(env::GymEnv, action) |
43 | | - ob, reward, done, information = env.gymenv[:step](action) |
| 43 | + ob, reward, done, information = env.gymenv.step(action) |
44 | 44 | return ob, reward, done, information |
45 | 45 | end |
46 | 46 |
|
47 | | -close!(env::GymEnv) = env.gymenv[:close]() |
| 47 | +close!(env::GymEnv) = env.gymenv.close() |
48 | 48 |
|
49 | | -seed!(env::GymEnv, seed=nothing) = env.gymenv[:seed](seed) |
| 49 | +seed!(env::GymEnv, seed=nothing) = env.gymenv.seed(seed) |
0 commit comments