Skip to content
This repository was archived by the owner on Aug 11, 2023. It is now read-only.

Commit 9e60c91

Browse files
github-actions[bot]github-actions[bot]
authored
Format .jl files (#111)
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent b3ac056 commit 9e60c91

File tree

10 files changed

+26
-23
lines changed

10 files changed

+26
-23
lines changed

‎src/base.jl

Lines changed: 16 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ function test_interfaces!(env)
115115
@testset "SingleAgent" begin
116116
if NumAgentStyle(env) === SINGLE_AGENT
117117
reset!(env)
118-
total_reward = 0.
118+
total_reward = 0.0
119119
while !is_terminated(env)
120120
if StateStyle(env) isa Tuple
121121
for ss in StateStyle(env)
@@ -272,7 +272,7 @@ end
272272

273273
using IntervalSets
274274

275-
Random.rand(s::Union{Interval, Array{<:Interval}}) = rand(Random.GLOBAL_RNG, s)
275+
Random.rand(s::Union{Interval,Array{<:Interval}}) = rand(Random.GLOBAL_RNG, s)
276276

277277
function Random.rand(rng::AbstractRNG, s::Interval)
278278
rand(rng) * (s.right - s.left) + s.left
@@ -292,7 +292,7 @@ struct WorldSpace{T} end
292292

293293
WorldSpace() = WorldSpace{Any}()
294294

295-
Base.in(x, ::WorldSpace{T}) where T = x isa T
295+
Base.in(x, ::WorldSpace{T}) where {T} = x isa T
296296

297297
#####
298298
# ZeroTo
@@ -305,16 +305,16 @@ Similar to `Base.OneTo`. Useful when wrapping third-party environments.
305305
"""
306306
struct ZeroTo{T<:Integer} <: AbstractUnitRange{T}
307307
stop::T
308-
ZeroTo{T}(n) where {T<:Integer} = new(max(zero(T)-one(T),n))
308+
ZeroTo{T}(n) where {T<:Integer} = new(max(zero(T) - one(T), n))
309309
end
310310

311311
ZeroTo(n::T) where {T<:Integer} = ZeroTo{T}(n)
312312

313313
Base.show(io::IO, r::ZeroTo) = print(io, "ZeroTo(", r.stop, ")")
314-
Base.length(r::ZeroTo{T}) where T = T(r.stop + one(r.stop))
315-
Base.first(r::ZeroTo{T}) where T = zero(r.stop)
314+
Base.length(r::ZeroTo{T}) where {T} = T(r.stop + one(r.stop))
315+
Base.first(r::ZeroTo{T}) where {T} = zero(r.stop)
316316

317-
function getindex(v::ZeroTo{T}, i::Integer) where T
317+
function getindex(v::ZeroTo{T}, i::Integer) where {T}
318318
Base.@_inline_meta
319319
@boundscheck ((i >= 0) & (i <= v.stop)) || throw_boundserror(v, i)
320320
convert(T, i)
@@ -349,7 +349,8 @@ function weighted_sample(rng::AbstractRNG, wv)
349349
end
350350
end
351351

352-
Random.rand(rng::AbstractRNG, s::AbstractVector{<:ActionProbPair}) = s[weighted_sample(rng, (x.prob for x in s))]
352+
Random.rand(rng::AbstractRNG, s::AbstractVector{<:ActionProbPair}) =
353+
s[weighted_sample(rng, (x.prob for x in s))]
353354

354355
(env::AbstractEnv)(a::ActionProbPair) = env(a.action)
355356

@@ -368,15 +369,16 @@ end
368369

369370
Random.rand(s::Space) = rand(Random.GLOBAL_RNG, s)
370371

371-
Random.rand(rng::AbstractRNG, s::Space) = map(s.s) do x
372-
rand(rng, x)
373-
end
372+
Random.rand(rng::AbstractRNG, s::Space) =
373+
map(s.s) do x
374+
rand(rng, x)
375+
end
374376

375-
Random.rand(rng::AbstractRNG, s::Space{<:Dict}) = Dict(k=>rand(rng,v) for (k,v) in s.s)
377+
Random.rand(rng::AbstractRNG, s::Space{<:Dict}) = Dict(k => rand(rng, v) for (k, v) in s.s)
376378

377379
function Base.in(X, S::Space)
378380
if length(X) == length(S.s)
379-
for (x,s) in zip(X, S.s)
381+
for (x, s) in zip(X, S.s)
380382
if x ∉ s
381383
return false
382384
end
@@ -398,4 +400,4 @@ function Base.in(X::Dict, S::Space{<:Dict})
398400
else
399401
return false
400402
end
401-
end
403+
end

‎src/examples/TicTacToeEnv.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,8 @@ current_player(env::TicTacToeEnv) = env.player
7575
players(env::TicTacToeEnv) = (CROSS, NOUGHT)
7676

7777
state(env::TicTacToeEnv, ::Observation{BitArray{3}}, p) = env.board
78-
state_space(env::TicTacToeEnv, ::Observation{BitArray{3}}, p) = Space(fill(false..true, 3, 3, 3))
78+
state_space(env::TicTacToeEnv, ::Observation{BitArray{3}}, p) =
79+
Space(fill(false..true, 3, 3, 3))
7980
state(env::TicTacToeEnv, ::Observation{Int}, p) = get_tic_tac_toe_state_info()[env].index
8081
state_space(env::TicTacToeEnv, ::Observation{Int}, p) =
8182
Base.OneTo(length(get_tic_tac_toe_state_info()))

‎test/examples/kuhn_poker.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,6 @@
22

33
env = KuhnPokerEnv()
44

5-
RLBase.test_interfaces!(env)
5+
RLBase.test_interfaces!(env)
66

77
end

‎test/examples/monty_hall_problem.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
rng = StableRNG(123)
44
env = MontyHallEnv(; rng = rng)
55

6-
RLBase.test_interfaces!(env)
6+
RLBase.test_interfaces!(env)
77

88
n_win_car = 0
99
N = 50_000

‎test/examples/multi_arm_bandits.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
env = MultiArmBanditsEnv(; rng = rng)
55
rewards = []
66

7-
RLBase.test_interfaces!(env)
7+
RLBase.test_interfaces!(env)
88

99
N = 50_000
1010
for _ in 1:N

‎test/examples/random_walk_1d.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
end_rewards = 3 => 5
44
env = RandomWalk1D(; rewards = end_rewards)
55

6-
RLBase.test_interfaces!(env)
6+
RLBase.test_interfaces!(env)
77

88
rng = StableRNG(123)
99
N = 50_000

‎test/examples/rock_paper_scissors.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
rng = StableRNG(123)
44
env = RockPaperScissorsEnv()
55

6-
RLBase.test_interfaces!(env)
6+
RLBase.test_interfaces!(env)
77

88
rewards = [[], []]
99
for _ in 1:50_000

‎test/examples/tic_tac_toe.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
env = TicTacToeEnv()
44

5-
RLBase.test_interfaces!(env)
5+
RLBase.test_interfaces!(env)
66

77
@test length(state_space(env, Observation{Int}())) == 5478
88

‎test/examples/tiger_problem_env.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
obs_prob = 0.85
55
env = TigerProblemEnv(; rng = rng, obs_prob = obs_prob)
66

7-
RLBase.test_interfaces!(env)
7+
RLBase.test_interfaces!(env)
88

99
rewards = []
1010
for _ in 1:50_000

‎test/examples/tiny_hanabi.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,6 @@
22

33
env = TinyHanabiEnv()
44

5-
RLBase.test_interfaces!(env)
5+
RLBase.test_interfaces!(env)
66

77
end

0 commit comments

Comments
 (0)