diff --git a/miplearn/__init__.py b/miplearn/__init__.py
index c1240b6..d522eaa 100644
--- a/miplearn/__init__.py
+++ b/miplearn/__init__.py
@@ -7,5 +7,5 @@ from .component import Component
 from .instance import Instance
 from .solvers import LearningSolver
 from .benchmark import BenchmarkRunner
-from .warmstart import WarmStartComponent
+from .warmstart import WarmStartComponent, KnnWarmStartPredictor, LogisticWarmStartPredictor
 from .branching import BranchPriorityComponent
\ No newline at end of file