fix: adding batch quantile loss to __all__

Signed-off-by: Lee, Kin Long Kelvin <[email protected]>
IntelLabs · Oct 13, 2024 · 5fd7ca8 · 5fd7ca8
1 parent a475867
commit 5fd7ca8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/matsciml/models/losses.py b/matsciml/models/losses.py
@@ -6,7 +6,7 @@
 from torch import nn
 
 
-__all__ = ["AtomWeightedL1", "AtomWeightedMSE"]
+__all__ = ["AtomWeightedL1", "AtomWeightedMSE", "BatchQuantileLoss"]
 
 
 class AtomWeightedL1(nn.Module):