remove lr_schedulers from main module

webis-de · Nov 21, 2024 · 3ba91b1 · 3ba91b1
1 parent 76be876
commit 3ba91b1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git 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
@@ -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",