Fixed typehints for make_(train_validation)_dataloader

graphnet-team · Sep 19, 2023 · 368b9a8 · 368b9a8
1 parent 4af2872
commit 368b9a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphnet/training/utils.py b/src/graphnet/training/utils.py
@@ -22,7 +22,7 @@
 def collate_fn(graphs: List[Data]) -> Batch:
     """Remove graphs with less than two DOM hits.
 
-    Should not occur in "production.
+    Should not occur in "productio"n.
     """
     graphs = [g for g in graphs if g.n_pulses > 1]
     return Batch.from_data_list(graphs)