Skip to content
This repository was archived by the owner on May 6, 2021. It is now read-only.

Commit d4f6c7d

Browse files
authored
add doc (#41)
1 parent 5761cdf commit d4f6c7d

File tree

4 files changed

+58
-0
lines changed

4 files changed

+58
-0
lines changed

src/environments/classic_control/cartpole.jl

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,19 @@ end
3535
Base.show(io::IO, env::CartPoleEnv{T}) where {T} =
3636
print(io, "CartPoleEnv{$T}($(env.params))")
3737

38+
"""
39+
CartPoleEnv(;kwargs...)
40+
41+
# Keyword arguments
42+
- `T = Float64`
43+
- `gravity = T(9.8)`
44+
- `masscart = T(1.0)`
45+
- `masspole = T(0.1)`
46+
- `halflength = T(0.5)`
47+
- `forcemag = T(10.0)`
48+
- `max_steps = 200`
49+
- `seed = nothing`
50+
"""
3851
function CartPoleEnv(;
3952
T = Float64,
4053
gravity = T(9.8),

src/environments/classic_control/mountain_car.jl

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,23 @@ mutable struct MountainCarEnv{A,T,R<:AbstractRNG} <: AbstractEnv
4747
rng::R
4848
end
4949

50+
"""
51+
MountainCarEnv(;kwargs...)
52+
53+
# Keyword arguments
54+
55+
- `T = Float64`
56+
- `continuous = false`
57+
- `seed = nothing`
58+
- `min_pos = -1.2`
59+
- `max_pos = 0.6`
60+
- `max_speed = 0.07`
61+
- `goal_pos = 0.5`
62+
- `max_steps = 200`
63+
- `goal_velocity = 0.0`
64+
- `power = 0.001`
65+
- `gravity = 0.0025`
66+
"""
5067
function MountainCarEnv(; T = Float64, continuous = false, seed = nothing, kwargs...)
5168
if continuous
5269
params = MountainCarEnvParams(; goal_pos = 0.45, power = 0.0015, T = T, kwargs...)

src/environments/classic_control/pendulum.jl

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,21 @@ mutable struct PendulumEnv{T,R<:AbstractRNG} <: AbstractEnv
2323
reward::T
2424
end
2525

26+
"""
27+
PwendulumEnv(;kwargs...)
28+
29+
# Keyword arguments
30+
31+
- `T = Float64`
32+
- `max_speed = T(8)`
33+
- `max_torque = T(2)`
34+
- `g = T(10)`
35+
- `m = T(1)`
36+
- `l = T(1)`
37+
- `dt = T(0.05)`
38+
- `max_steps = 200`
39+
- `seed = nothing`
40+
"""
2641
function PendulumEnv(;
2742
T = Float64,
2843
max_speed = T(8),

src/environments/non_interactive/pendulum.jl

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,19 @@ mutable struct PendulumNonInteractiveEnv{
3434
rng::R
3535
end
3636

37+
"""
38+
PendulumNonInteractiveEnv(;kwargs...)
39+
40+
# Keyword arguments
41+
42+
- `float_type = Float64`
43+
- `gravity = 9.8`
44+
- `length = 2.0`
45+
- `mass = 1.0`
46+
- `step_size = 0.01`
47+
- `maximum_time = 10.0`
48+
- `seed = nothing`
49+
"""
3750
function PendulumNonInteractiveEnv(;
3851
float_type = Float64,
3952
gravity = 9.8,

0 commit comments

Comments
 (0)