diff --git a/test/reinforcement_learning_flattening.jl b/test/reinforcement_learning_flattening.jl index 2fdf4739..ce7b5d92 100644 --- a/test/reinforcement_learning_flattening.jl +++ b/test/reinforcement_learning_flattening.jl @@ -7,7 +7,7 @@ using DataFrames using CSV using ModelingToolkit: getp -@testset "RL test" begin +@testset "RL test with save" begin t_trial = 2 # ms time_block_dur = 0.01 # ms N_trials = 3 @@ -69,7 +69,7 @@ using ModelingToolkit: getp init_params_idxs_other_params = params_at(idxs_other_params) env = ClassificationEnvironment(stim; name=:env, namespace=global_ns) - run_experiment!(agent, env, "/Users/achesebro/Downloads/"; alg=Vern7(), reltol=1e-9,abstol=1e-9) + run_experiment!(agent, env, "./"; alg=Vern7(), reltol=1e-9,abstol=1e-9) final_params = reduce(vcat, agent.problem.p) # At least some weights need to be different.