@@ -21,7 +21,7 @@ class IncorrectStepReasoningVariant:
2121 name : str
2222 regenerate_steps : Optional [bool ] = True
2323 generate_and_rate_alternative_steps : Optional [bool ] = True
24- rewrite_steps : Optional [bool ] = True
24+ rewrite_step : Optional [bool ] = True
2525 justification : Optional [bool ] = True
2626
2727 def asdict (self ) -> Dict [str , Any ]:
@@ -30,8 +30,8 @@ def asdict(self) -> Dict[str, Any]:
3030 actions .append ("regenerateSteps" )
3131 if self .generate_and_rate_alternative_steps :
3232 actions .append ("generateAndRateAlternativeSteps" )
33- if self .rewrite_steps :
34- actions .append ("rewriteSteps " )
33+ if self .rewrite_step :
34+ actions .append ("rewriteStep " )
3535 if self .justification :
3636 actions .append ("justification" )
3737 return {"id" : self .id , "name" : self .name , "actions" : actions }
@@ -46,7 +46,7 @@ def from_dict(
4646 regenerate_steps = "regenerateSteps" in dictionary .get ("actions" , []),
4747 generate_and_rate_alternative_steps = "generateAndRateAlternativeSteps"
4848 in dictionary .get ("actions" , []),
49- rewrite_steps = "rewriteSteps " in dictionary .get ("actions" , []),
49+ rewrite_step = "rewriteStep " in dictionary .get ("actions" , []),
5050 justification = "justification" in dictionary .get ("actions" , []),
5151 )
5252
@@ -183,11 +183,11 @@ def reset_generate_and_rate_alternative_steps(self):
183183 """
184184 self .definition .variants .incorrect_step .generate_and_rate_alternative_steps = False
185185
186- def reset_rewrite_steps (self ):
186+ def reset_rewrite_step (self ):
187187 """
188188 For live models, will require labelers to rewrite the conversation
189189 """
190- self .definition .variants .incorrect_step .rewrite_steps = False
190+ self .definition .variants .incorrect_step .rewrite_step = False
191191
192192 def reset_justification (self ):
193193 """
0 commit comments