From db0a2d567b4429c7288a1fbb4c7651107a9f92a5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Thu, 20 Apr 2023 15:22:46 +0200
Subject: [PATCH] fixes #25

---
 src/LarvaDatasets.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/LarvaDatasets.jl b/src/LarvaDatasets.jl
index b531b73..0fd33be 100644
--- a/src/LarvaDatasets.jl
+++ b/src/LarvaDatasets.jl
@@ -597,7 +597,8 @@ function write_larva_dataset_hdf5(output_dir::String,
         sample_sizes, total_sample_size = thresholdedcounts(counts)
     end
     # ensure label order is preserved
-    sample_sizes = OrderedDict((label => sample_sizes[label]) for label in labels if label in keys(sample_sizes))
+    labels′= isnothing(includeall) ? labels : push!(collect(labels), includeall)
+    sample_sizes = OrderedDict((label => sample_sizes[label]) for label in labels′ if label in keys(counts))
     #
     @info "Sample sizes (observed, selected):" [Symbol(label) => (get(counts, label, 0), get(sample_sizes, label, 0)) for label in labels]...
     date = Dates.format(Dates.now(), "yyyy_mm_dd")
-- 
GitLab