From ea420224f9594f17af7c323d54c26b1e9b876928 Mon Sep 17 00:00:00 2001 From: Mo8it Date: Wed, 6 Apr 2022 17:07:04 +0200 Subject: [PATCH] Fix reward_discount_analysis --- analysis/Q_matrix_latex_export.jl | 6 +++--- analysis/reward_discount_analysis.jl | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/analysis/Q_matrix_latex_export.jl b/analysis/Q_matrix_latex_export.jl index 73b738c..5d96479 100644 --- a/analysis/Q_matrix_latex_export.jl +++ b/analysis/Q_matrix_latex_export.jl @@ -4,10 +4,10 @@ using PrettyTables: pretty_table using ReCo: ReCo function latex_table( - dataframe::DataFrames.DataFrame, filename::String; path::String = "exports/$filename" + dataframe::DataFrames.DataFrame, filename::String; path::String="exports/$filename" ) open(path, "w") do f - pretty_table(f, dataframe; backend = :latex, nosubheader = true, alignment = :c) + pretty_table(f, dataframe; backend=:latex, nosubheader=true, alignment=:c) end return nothing @@ -29,7 +29,7 @@ function export_q_matrix(env_helper::ReCo.RL.EnvHelper, filename_without_extensi table[:, col] ./= sum(table[:, col]) end - table .= round.(table, digits = 2) + table .= round.(table, digits=2) state_spaces_labels = env_helper.shared.env.shared.state_spaces_labels states = AbstractString[] diff --git a/analysis/reward_discount_analysis.jl b/analysis/reward_discount_analysis.jl index a9a450b..a04be0f 100644 --- a/analysis/reward_discount_analysis.jl +++ b/analysis/reward_discount_analysis.jl @@ -11,8 +11,8 @@ function run_reward_discount_processes(γs::AbstractVector) Threads.@threads for γ_ind in 1:n_γs γ = γs[γ_ind] - env_helper, rl_dir = ReCo.run_rl(; - EnvType=ReCo.OriginEnv, + env_helper, rl_dir = ReCo.run_rl( + ReCo.OriginEnv; n_episodes=400, episode_duration=15.0, n_particles=150,