From c94f40a6e34e6c3b3c5dcf5b2fe730797c935506 Mon Sep 17 00:00:00 2001 From: Isaac Miller Date: Sun, 3 Mar 2024 23:54:30 -0600 Subject: [PATCH] Move ensemble import to teleprompt --- dspy/teleprompt/__init__.py | 1 + tests/teleprompt/test_ensemble.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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):