diff --git a/lightning_ir/__init__.py b/lightning_ir/__init__.py
index 14fe378..780d2b8 100644
--- a/lightning_ir/__init__.py
+++ b/lightning_ir/__init__.py
@@ -42,7 +42,6 @@
     TupleDataset,
 )
 from .lightning_utils import (
-    LR_SCHEDULERS,
     ConstantLRSchedulerWithLinearWarmup,
     GenericConstantSchedulerWithLinearWarmup,
     GenericConstantSchedulerWithQuadraticWarmup,
@@ -180,7 +179,6 @@
     "LightningIRWandbLogger",
     "LinearLRSchedulerWithLinearWarmup",
     "LocalizedContrastiveEstimation",
-    "LR_SCHEDULERS",
     "QueryDataset",
     "QuerySample",
     "RankCallback",
diff --git a/lightning_ir/lightning_utils/__init__.py b/lightning_ir/lightning_utils/__init__.py
index b2b6268..b7ccca7 100644
--- a/lightning_ir/lightning_utils/__init__.py
+++ b/lightning_ir/lightning_utils/__init__.py
@@ -1,10 +1,5 @@
 from .callbacks import IndexCallback, RankCallback, ReRankCallback, SearchCallback
-from .lr_schedulers import (
-    LR_SCHEDULERS,
-    ConstantLRSchedulerWithLinearWarmup,
-    LinearLRSchedulerWithLinearWarmup,
-    WarmupLRScheduler,
-)
+from .lr_schedulers import ConstantLRSchedulerWithLinearWarmup, LinearLRSchedulerWithLinearWarmup, WarmupLRScheduler
 from .schedulers import (
     GenericConstantSchedulerWithLinearWarmup,
     GenericConstantSchedulerWithQuadraticWarmup,
@@ -18,7 +13,6 @@
     "GenericLinearSchedulerWithLinearWarmup",
     "IndexCallback",
     "LinearLRSchedulerWithLinearWarmup",
-    "LR_SCHEDULERS",
     "RankCallback",
     "ReRankCallback",
     "SearchCallback",