fix some step/reset usage

Author: rejuvyesh

environments/ant/methods/env.jl

@@ -76,7 +76,7 @@ function ant(;
     return env
 end
 
-function step(env::Environment{Ant}, x, u; 
+function Base.step(env::Environment{Ant}, x, u; 
     gradients=false,
     attitude_decompress=false)
 
@@ -144,7 +144,7 @@ function step(env::Environment{Ant}, x, u;
 end
 
 # TODO add random noise
-function reset(env::Environment{Ant}; 
+function Base.reset(env::Environment{Ant}; 
     x=nothing)
 
     initialize!(env.mechanism, type2symbol(Ant))

environments/halfcheetah/methods/env.jl

@@ -77,7 +77,7 @@ function halfcheetah(;
     return env
 end
 
-function reset(env::Environment{HalfCheetah}; 
+function Base.reset(env::Environment{HalfCheetah}; 
     x=nothing, reset_noise_scale=0.1)
     if x != nothing
         env.state .= x

environments/pendulum/methods/env.jl

@@ -70,7 +70,7 @@ function pendulum(;
     return env
 end
 
-function reset(env::Environment{Pendulum}; 
+function Base.reset(env::Environment{Pendulum}; 
     x=nothing)
 
     initialize!(env.mechanism, :pendulum)
@@ -99,7 +99,7 @@ function get_observation(env::Environment{Pendulum})
     end
 end
 
-function step(env::Environment{Pendulum}, x, u; 
+function Base.step(env::Environment{Pendulum}, x, u; 
     gradients=false,
     attitude_decompress=false)
     mechanism = env.mechanism

environments/rexhopper/methods/env.jl

@@ -85,7 +85,7 @@ function rexhopper(;
     return env
 end
 
-function reset(env::Environment{RexHopper}; 
+function Base.reset(env::Environment{RexHopper}; 
     x=nothing)
 
     if x != nothing

environments/walker/methods/env.jl

@@ -78,7 +78,7 @@ function walker(;
     return env
 end
 
-function reset(env::Environment{Walker}; x=nothing, reset_noise_scale = 0.005)
+function Base.reset(env::Environment{Walker}; x=nothing, reset_noise_scale = 0.005)
     if x != nothing
         env.state .= x
     else