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

feat: Add separate maternal and paternal grandparents to family relationships

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-27 20:41:13 +01:00
parent 9f878e0244
commit 84c0107ce0
1 changed files with 18 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

25
reasoning_gym/graphs/family_relationships.py
View file

@ -233,18 +233,29 @@ class FamilyRelationshipsDataset(ProceduralDataset):
# Create ID counter
id_counter = count()
# Create grandparents generation
grandfather = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
grandmother = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
grandfather.add_spouse(grandmother)
family.update([grandfather, grandmother])
# Create paternal grandparents generation
grandfather_of_father = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
grandmother_of_father = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
grandfather_of_father.add_spouse(grandmother_of_father)
family.update([grandfather_of_father, grandmother_of_father])
# Create maternal grandparents generation
grandfather_of_mother = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
grandmother_of_mother = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
grandfather_of_mother.add_spouse(grandmother_of_mother)
family.update([grandfather_of_mother, grandmother_of_mother])
# Create parents
father = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
mother = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
father.add_spouse(mother)
grandfather.add_child(father)
grandmother.add_child(father)
# Link parents to their respective parents
grandfather_of_father.add_child(father)
grandmother_of_father.add_child(father)
grandfather_of_mother.add_child(mother)
grandmother_of_mother.add_child(mother)
family.update([father, mother])
# Add children