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

feat: Randomly assign children to multiple couples in family generation

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-27 20:54:58 +01:00
parent a00bab6e21
commit 277e65a7e0
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
reasoning_gym/graphs/family_relationships.py
View file

@ -272,15 +272,18 @@ class FamilyRelationshipsDataset(ProceduralDataset):
family.update([father, mother, uncle, aunt_by_marriage, aunt, uncle_by_marriage])
# Add children
# Add children, randomly assigned to couples
couples = [(father, mother), (uncle, aunt_by_marriage), (aunt, uncle_by_marriage)]
while len(family) < family_size:
gender = rng.choice([Gender.MALE, Gender.FEMALE])
name = get_name(gender)
if not name:
break
child = Person(name, gender, next(id_counter))
father.add_child(child)
mother.add_child(child)
# Randomly choose parents for this child
parents = rng.choice(couples)
parents[0].add_child(child) # Add to father/uncle/aunt
parents[1].add_child(child) # Add to mother/aunt_by_marriage/uncle_by_marriage
family.add(child)
return family