From 11c00becb9ef5c9b17e388a28178c0825047ac85 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Mon, 13 Feb 2023 19:52:30 +0100
Subject: [PATCH] fixes
 https://gitlab.pasteur.fr/nyx/larvatagger.jl/-/issues/98

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

diff --git a/src/LarvaDatasets.jl b/src/LarvaDatasets.jl
index 22d590c..70d9183 100644
--- a/src/LarvaDatasets.jl
+++ b/src/LarvaDatasets.jl
@@ -336,6 +336,9 @@ function write_larva_dataset_hdf5(path, counts, files, refs, nsteps_before, nste
         end
         attributes(g)["n_samples"] = sampleid
         # extension
+        if !isnothing(includeall)
+            delete!(counts, includeall)
+        end
         h5["labels"] = collect(keys(counts))
         h5["label_counts"] = collect(values(counts))
         #h5["files"] = [f.source for f in files]
-- 
GitLab