Files
courses/evaluations/05_prompt_foo_code_graded_animals/transform.py
2024-09-04 16:45:46 -06:00

10 lines
283 B
Python

def get_transform(output, context):
if "<thinking>" in output:
try:
return output.split("<answer>")[1].split("</answer>")[0].strip()
except Exception as e:
print(f"Error in get_transform: {e}")
return output
return output