From 5fd7ca88d8a3ce141502b910b9f148e6272898e7 Mon Sep 17 00:00:00 2001
From: "Lee, Kin Long Kelvin" <kin.long.kelvin.lee@intel.com>
Date: Sun, 13 Oct 2024 13:35:51 -0700
Subject: [PATCH] fix: adding batch quantile loss to __all__

Signed-off-by: Lee, Kin Long Kelvin <kin.long.kelvin.lee@intel.com>
---
 matsciml/models/losses.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/matsciml/models/losses.py b/matsciml/models/losses.py
index 207da6d2..75ce57e0 100644
--- 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):