reformat using black

sfluegel05 · sfluegel05 · commit 0c2b99d6f65d · 2025-03-28T10:57:12.000+01:00
diff --git a/chebai/loss/bce_weighted.py b/chebai/loss/bce_weighted.py
@@ -61,10 +61,14 @@ def set_pos_weight(self, input: torch.Tensor) -> None:
             )
             complete_labels = torch.concat(
                 [
-                    torch.stack([
-                        torch.Tensor(row["labels"]) for row in
-                        self.data_extractor.load_processed_data(filename=file_name)
-                    ])
+                    torch.stack(
+                        [
+                            torch.Tensor(row["labels"])
+                            for row in self.data_extractor.load_processed_data(
+                                filename=file_name
+                            )
+                        ]
+                    )
                     for file_name in self.data_extractor.processed_file_names
                 ]
             )

Original file line number	Diff line number	Diff line change
`@@ -61,10 +61,14 @@ def set_pos_weight(self, input: torch.Tensor) -> None:`
`61`	`61`	`)`
`62`	`62`	`complete_labels = torch.concat(`
`63`	`63`	`[`
`64`		`- torch.stack([`
`65`		`- torch.Tensor(row["labels"]) for row in`
`66`		`- self.data_extractor.load_processed_data(filename=file_name)`
`67`		`- ])`
	`64`	`+ torch.stack(`
	`65`	`+ [`
	`66`	`+ torch.Tensor(row["labels"])`
	`67`	`+ for row in self.data_extractor.load_processed_data(`
	`68`	`+ filename=file_name`
	`69`	`+ )`
	`70`	`+ ]`
	`71`	`+ )`
`68`	`72`	`for file_name in self.data_extractor.processed_file_names`
`69`	`73`	`]`
`70`	`74`	`)`