File tree 1 file changed +4
-1
lines changed
evals/elsuite/schelling_point
1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 9
9
from evals .elsuite .schelling_point .prompts import sys_prompts_ci , sys_prompts_no_ci
10
10
from evals .elsuite .schelling_point .utils import get_response
11
11
from evals .eval import Eval
12
+ from evals .solvers .solver import Solver
12
13
13
14
14
15
class SchellingPoint (Eval ):
@@ -22,6 +23,9 @@ def __init__(
22
23
* args ,
23
24
** kwargs ,
24
25
):
26
+ if any ([isinstance (completion_fn , Solver ) for completion_fn in completion_fns ]):
27
+ raise NotImplementedError ("SchellingPoint does not currently support Solvers" )
28
+
25
29
super ().__init__ (completion_fns , * args , ** kwargs )
26
30
27
31
random .seed (seed )
@@ -34,7 +38,6 @@ def __init__(
34
38
35
39
self .temperature = temperature
36
40
37
- self .completion_fns = completion_fns
38
41
if len (self .completion_fns ) == 1 :
39
42
self .completion_fns = self .completion_fns * n_copies
40
43
assert len (self .completion_fns ) == n_copies , "Must provide n_copies completion_fns"
You can’t perform that action at this time.
0 commit comments