add current_t kwarg for the (full) constructor (#86)

JuliaReinforcementLearning · Oct 13, 2020 · 6ee5216 · 6ee5216
1 parent 670b9e3
commit 6ee5216
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/implementations/environments.jl b/src/implementations/environments.jl
@@ -159,6 +159,8 @@ mutable struct MaxTimeoutEnv{E<:AbstractEnv} <: AbstractEnv
     current_t::Int
 end
 
+MaxTimeoutEnv(env::E, max_t::Int; current_t::Int = 1) where E<:AbstractEnv = MaxTimeoutEnv(E, max_t, current_t)
+
 function (env::MaxTimeoutEnv)(args...; kwargs...)
     env.env(args...; kwargs...)
     env.current_t = env.current_t + 1