Skip to content
Snippets Groups Projects
Commit 93b42dca authored by François  LAURENT's avatar François LAURENT
Browse files

fixes #6

parent a1282478
No related branches found
No related tags found
No related merge requests found
...@@ -71,7 +71,7 @@ class MaggotTrainer: ...@@ -71,7 +71,7 @@ class MaggotTrainer:
def pad(self, target_t, defined_t, data): def pad(self, target_t, defined_t, data):
if data.shape[0] == 1: if data.shape[0] == 1:
return data return np.repeat(data, len(target_t), axis=0)
else: else:
head = searchsortedfirst(target_t, defined_t[0]) head = searchsortedfirst(target_t, defined_t[0])
tail = len(target_t) - (searchsortedlast(target_t, defined_t[-1]) + 1) tail = len(target_t) - (searchsortedlast(target_t, defined_t[-1]) + 1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment