From 93b42dcadd42e8e47094a5d4c44604398659b7da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Tue, 7 Feb 2023 14:43:50 +0100
Subject: [PATCH] fixes #6
---
src/maggotuba/models/trainers.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/maggotuba/models/trainers.py b/src/maggotuba/models/trainers.py
index 154d6f1..6852e40 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)
--
GitLab