Skip to content

Commit a506f0c

Browse files
committed
Add NotImplementedError if user tries to use a solver
1 parent b01c56d commit a506f0c

File tree

1 file changed

+4
-1
lines changed
  • evals/elsuite/schelling_point

1 file changed

+4
-1
lines changed

Diff for: evals/elsuite/schelling_point/eval.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from evals.elsuite.schelling_point.prompts import sys_prompts_ci, sys_prompts_no_ci
1010
from evals.elsuite.schelling_point.utils import get_response
1111
from evals.eval import Eval
12+
from evals.solvers.solver import Solver
1213

1314

1415
class SchellingPoint(Eval):
@@ -22,6 +23,9 @@ def __init__(
2223
*args,
2324
**kwargs,
2425
):
26+
if any([isinstance(completion_fn, Solver) for completion_fn in completion_fns]):
27+
raise NotImplementedError("SchellingPoint does not currently support Solvers")
28+
2529
super().__init__(completion_fns, *args, **kwargs)
2630

2731
random.seed(seed)
@@ -34,7 +38,6 @@ def __init__(
3438

3539
self.temperature = temperature
3640

37-
self.completion_fns = completion_fns
3841
if len(self.completion_fns) == 1:
3942
self.completion_fns = self.completion_fns * n_copies
4043
assert len(self.completion_fns) == n_copies, "Must provide n_copies completion_fns"

0 commit comments

Comments
 (0)