File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
auto_graph_of_thoughts/language_model Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 42
42
}
43
43
44
44
45
- def _intersect_list_correctly (prompt : Prompt , state : State ) -> State :
45
+ def _intersect_set_correctly (prompt : Prompt , state : State ) -> State :
46
46
if 'set1' not in state or 'set2' not in state :
47
47
return {
48
48
'intersection' : []
@@ -129,7 +129,7 @@ def create_simulated_realistic_chat_gpt_intersect_set(seed: int) -> SimulatedLan
129
129
simulated_behaviors = [
130
130
SimulatedLanguageModelBehavior (
131
131
prompt = op_intersect .prompt ,
132
- mocked_correct_behavior = _intersect_list_correctly ,
132
+ mocked_correct_behavior = _intersect_set_correctly ,
133
133
mocked_incorrect_behavior = _intersect_set_incorrectly ,
134
134
probability = _get_intersect_set_probability
135
135
),
@@ -157,7 +157,7 @@ def create_simulated_deterministic_chat_gpt_intersect_set(seed: int) -> Simulate
157
157
simulated_behaviors = [
158
158
SimulatedLanguageModelBehavior (
159
159
prompt = op_intersect .prompt ,
160
- mocked_correct_behavior = _intersect_list_correctly ,
160
+ mocked_correct_behavior = _intersect_set_correctly ,
161
161
mocked_incorrect_behavior = _intersect_set_incorrectly ,
162
162
probability = lambda p , s : 1.0 if _get_intersect_set_probability (p , s ) == 1.0 else 0.0
163
163
),
You can’t perform that action at this time.
0 commit comments