diff --git a/dspy/teleprompt/__init__.py b/dspy/teleprompt/__init__.py index 61425e5b9b..a1088f2df3 100644 --- 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 index 292176af4f..54ca2708db 100644 --- 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):