black

MaloOLIVIER · Jan 2, 2025 · 6a1a800 · 6a1a800
1 parent de13dc2
commit 6a1a800
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/hungarian_net/lightning_datamodules/hungarian_datamodule.py b/hungarian_net/lightning_datamodules/hungarian_datamodule.py
@@ -161,10 +161,6 @@ def compute_weighted_accuracy(self, n1star: int, n0star: int) -> float:
 
         return WA
 
-
-# TODO: factorize HungarianDataset in HungarianDataModule
-
-
 class HungarianDataModule(LightningDataModule):
     """
     A PyTorch Lightning DataModule for managing data loading for the Hungarian Network.