Merge pull request #1884 from anthony-wss/develop

Gastron · web-flow · commit 14b7122cddd8 · 2023-03-22T13:31:14.000+02:00
fix batch_to_right
diff --git a/speechbrain/utils/data_utils.py b/speechbrain/utils/data_utils.py
@@ -418,7 +418,7 @@ def batch_pad_right(tensors: list, mode="constant", value=0):
         return tensors[0].unsqueeze(0), torch.tensor([1.0])
 
     if not (
-        any(
+        all(
             [tensors[i].ndim == tensors[0].ndim for i in range(1, len(tensors))]
         )
     ):

Original file line number	Diff line number	Diff line change
`@@ -418,7 +418,7 @@ def batch_pad_right(tensors: list, mode="constant", value=0):`
`418`	`418`	`return tensors[0].unsqueeze(0), torch.tensor([1.0])`
`419`	`419`
`420`	`420`	`if not (`
`421`		`- any(`
	`421`	`+ all(`
`422`	`422`	`[tensors[i].ndim == tensors[0].ndim for i in range(1, len(tensors))]`
`423`	`423`	`)`
`424`	`424`	`):`