From 91c07d67bc65e58c268ba5c40345f77b55b58dfa Mon Sep 17 00:00:00 2001 From: Olivier Labayle Date: Thu, 1 Feb 2024 11:47:06 +0100 Subject: [PATCH] add filter SAMPLE_IDS from estimators --- src/plots.jl | 4 ++-- test/plots.jl | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/plots.jl b/src/plots.jl index 251cd66..6fc665a 100644 --- a/src/plots.jl +++ b/src/plots.jl @@ -21,11 +21,11 @@ end function load_results(file_or_prefix; verbosity=0) verbosity > 0 && @info "Loading results." results = isfile(file_or_prefix) ? read_results_file(file_or_prefix) : read_results_files(file_or_prefix) - estimators = collect(keys(first(results))) + estimators = collect(key for key ∈ keys(first(results)) if key !== :SAMPLE_IDS) results_df = DataFrame([[r[id] for r in results] for id in 1:length(estimators)], estimators) for estimator in estimators results_df[!, Symbol(estimator, :_PVALUE)] = [pvalue(x) for x in results_df[!, estimator]] - end + end return results_df end diff --git a/test/plots.jl b/test/plots.jl index 174871b..dc36d27 100644 --- a/test/plots.jl +++ b/test/plots.jl @@ -8,7 +8,7 @@ TESTDIR = joinpath(pkgdir(TargeneCore), "test") include(joinpath(TESTDIR, "testutils.jl")) -@testset "Test " begin +@testset "Test generate_summary_plots" begin tmpdir = mktempdir() prefix = joinpath(tmpdir, "tmle_output") make_fake_outputs(; prefix=prefix)