update forwarding of play! method for RLBaseEnv

JuliaReinforcementLearning · Jul 23, 2021 · 66d5f7a · 66d5f7a
1 parent 689cbe3
commit 66d5f7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/rlbase.jl b/src/rlbase.jl
@@ -4,5 +4,5 @@ end
 
 Base.show(io::IO, mime::MIME"text/plain", env::RLBaseEnv{E}) where {E <: AbstractGridWorld} = show(io, mime, env.env)
 
-play!(terminal::REPL.Terminals.UnixTerminal, env::RLBaseEnv{E}; file_name::Union{Nothing, AbstractString} = nothing) where {E <: AbstractGridWorld} = play!(terminal, env.env, file_name = file_name)
-play!(env::RLBaseEnv{E}; file_name = nothing) where {E <: AbstractGridWorld} = play!(REPL.TerminalMenus.terminal, env.env, file_name = file_name)
+play!(terminal::REPL.Terminals.UnixTerminal, env::RLBaseEnv{E}, file_name::Union{Nothing, AbstractString}, frame_start_delimiter::Union{Nothing, AbstractString}) where {E <: AbstractGridWorld} = play!(terminal, env.env, file_name, frame_start_delimiter)
+play!(env::RLBaseEnv{E}; file_name = nothing, frame_start_delimiter = nothing) where {E <: AbstractGridWorld} = play!(REPL.TerminalMenus.terminal, env.env, file_name, frame_start_delimiter)