open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-24 17:05:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Template support recursive symbolics

Browse source
This commit is contained in:
EduardDurech 2025-02-08 18:31:29 +00:00
parent e320ba66dd
commit b3e61988c1
2 changed files with 23 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

26
reasoning_gym/core/template.py
View file

@ -21,11 +21,29 @@ def execute_template(template_str: str, parts: Dict[str, str | Callable], refs:
)
values[name] = result["question"]
executed_parts[name] = result["executed_parts"]
# Direct value generator
elif callable(part):
# Execute the callable
result = part(refs) if "refs" in part.__code__.co_varnames else part()
# Check if result is a template
if isinstance(result, dict) and "template" in result and "parts" in result:
# Handle nested template
nested_result = execute_template(
result["template"],
result["parts"],
refs
)
values[name] = nested_result["question"]
executed_parts[name] = nested_result["executed_parts"]
else:
# Direct value
values[name] = str(result)
executed_parts[name] = result
else:
# Direct value generator
value = part(refs) if "refs" in part.__code__.co_varnames else part()
values[name] = str(value)
executed_parts[name] = str(value) # Store value directly without nesting
# Static value
values[name] = str(part)
executed_parts[name] = part
return {
"question": template_str.format(**values),