diff --git a/src/LarvaDatasets.jl b/src/LarvaDatasets.jl index 22d590cc09a4509a7117207aea3d194e9b3da433..70d9183f397933c6fab6cc255ed244a95a1efc34 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]