sfix for issue

jcapp4 · jcapp4 · commit 7721d8c732fb · 2025-09-30T20:23:46.000+02:00
diff --git a/chebai/preprocessing/collate.py b/chebai/preprocessing/collate.py
@@ -130,7 +130,7 @@ def process_label_rows(self, labels: Tuple) -> torch.Tensor:
         """
         return pad_sequence(
             [
-                torch.tensor([v if v is not None else False for v in row])
+                torch.tensor([bool(v) if bool(v) is not None else False for v in row])
                 for row in labels
             ],
             batch_first=True,

Original file line number	Diff line number	Diff line change
`@@ -130,7 +130,7 @@ def process_label_rows(self, labels: Tuple) -> torch.Tensor:`
`130`	`130`	`"""`
`131`	`131`	`return pad_sequence(`
`132`	`132`	`[`
`133`		`- torch.tensor([v if v is not None else False for v in row])`
	`133`	`+ torch.tensor([bool(v) if bool(v) is not None else False for v in row])`
`134`	`134`	`for row in labels`
`135`	`135`	`],`
`136`	`136`	`batch_first=True,`