diff --git a/src/maggotuba/models/trainers.py b/src/maggotuba/models/trainers.py index 154d6f139282c0a52d452c9f9937d991afa32cae..6852e40dbc23d9b8d3ff2155333c6315f4928e12 100644 --- a/src/maggotuba/models/trainers.py +++ b/src/maggotuba/models/trainers.py @@ -71,7 +71,7 @@ class MaggotTrainer: def pad(self, target_t, defined_t, data): if data.shape[0] == 1: - return data + return np.repeat(data, len(target_t), axis=0) else: head = searchsortedfirst(target_t, defined_t[0]) tail = len(target_t) - (searchsortedlast(target_t, defined_t[-1]) + 1)