fix batching bug with LabelTensor

FilippoOlivo · FilippoOlivo · commit fc0a7e9fc6b7 · 2026-02-17T10:20:46.000+01:00
diff --git a/pina/_src/condition/data_manager.py b/pina/_src/condition/data_manager.py
@@ -119,7 +119,7 @@ def create_batch(items):
                     if isinstance(sample, LabelTensor)
                     else torch.stack
                 )
-                batch_data[k] = batch_fn(vals, dim=0)
+                batch_data[k] = batch_fn(vals)
             else:
                 batch_data[k] = sample
         return batch_data

Original file line number	Diff line number	Diff line change
`@@ -119,7 +119,7 @@ def create_batch(items):`
`119`	`119`	`if isinstance(sample, LabelTensor)`
`120`	`120`	`else torch.stack`
`121`	`121`	`)`
`122`		`- batch_data[k] = batch_fn(vals, dim=0)`
	`122`	`+ batch_data[k] = batch_fn(vals)`
`123`	`123`	`else:`
`124`	`124`	`batch_data[k] = sample`
`125`	`125`	`return batch_data`