Move ensemble import to teleprompt (stanfordnlp#542)

bakebrain · Mar 5, 2024 · ee64653 · ee64653
1 parent e4edd5b
commit ee64653
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dspy/teleprompt/__init__.py b/dspy/teleprompt/__init__.py
@@ -1,4 +1,5 @@
 from .bootstrap import *
+from .ensemble import *
 from .finetune import *
 from .knn_fewshot import *
 from .random_search import *

diff --git a/tests/teleprompt/test_ensemble.py b/tests/teleprompt/test_ensemble.py
@@ -1,6 +1,6 @@
 import pytest
 import dspy
-from dspy.teleprompt.ensemble import Ensemble
+from dspy.teleprompt import Ensemble
 
 
 class MockProgram(dspy.Module):