From f62b5deba4792d4b593b3d6ec4077411da44b5b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Mon, 28 Aug 2023 17:01:50 +0200
Subject: [PATCH] bugfix for balancing_strategy=nothing
---
src/LarvaDatasets.jl | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/LarvaDatasets.jl b/src/LarvaDatasets.jl
index 1c12916..8188429 100644
--- a/src/LarvaDatasets.jl
+++ b/src/LarvaDatasets.jl
@@ -540,7 +540,7 @@ Known issue: ASCII-compatible string attributes are ASCII encoded and deserializ
by the *h5py* Python library.
"""
function write_larva_dataset_hdf5(output_dir::String,
- input_data::Union{String, <:AbstractVector},
+ input_data::Union{Repository, String, <:AbstractVector},
window_length::Int=20;
labels::Union{Nothing, <:AbstractVector{String}}=nothing,
labelpointers::Union{Nothing, <:AbstractDict{String, Vector{Tuple{Int, Int, Int}}}}=nothing,
@@ -619,7 +619,7 @@ function new_write_larva_dataset_hdf5(output_dir, input_data;
@assert !isnothing(frameinterval)
window = TimeWindow(window_length * frameinterval, round(Int, 1 / frameinterval);
maggotuba_compatibility=past_future_extensions)
- index = if startswith(balancing_strategy, "max:")
+ index = if !isnothing(balancing_strategy) && startswith(balancing_strategy, "max:")
# `includeall` not supported
maxcount = parse(Int, balancing_strategy[5:end])
capacitysampling(labels, maxcount; seed=seed)
--
GitLab