open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add uncle, aunt & niece and nephew family relationships

Browse source
This commit is contained in:
Andreas Koepf 2025-01-27 21:19:48 +01:00
parent 277e65a7e0
commit c8219be6ae
2 changed files with 39 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

2
README.md
View file

@ -105,7 +105,7 @@ Available dataset names (which can be used with `create_dataset()`):
- `CaesarCipherDataset`: Encrypt/decrypt text using Caesar cipher with configurable rotation
- `LetterCountingDataset`: Count letter occurrences in text spans
- `NumberFilteringDataset`: Filter numbers based on comparison with threshold
- `NumberSortingDataset`: Sort lists of numbers in ascending or descending order
- `NumberSortingDataset`: Sort lists of numbers in ascending or descending order
- `WordSortingDataset`: Sort words in ascending or descending order using ASCII/Unicode ordering
- `LetterJumbleDataset`: Unscramble words that have had their letters randomly jumbled
- `SentenceReorderingDataset`: Reorder sentence after words in it have been randomly shuffled

65
reasoning_gym/graphs/family_relationships.py
View file

@ -239,41 +239,52 @@ class FamilyRelationshipsDataset(ProceduralDataset):
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])
if family_size > 6:
# 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])
couples = []
# 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)
# Create father's brother (uncle) and his wife
uncle = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
aunt_by_marriage = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
uncle.add_spouse(aunt_by_marriage)
# Create father's sister (aunt) and her husband
aunt = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
uncle_by_marriage = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
aunt.add_spouse(uncle_by_marriage)
# Link parents to their respective parents
grandfather_of_father.add_child(father)
grandmother_of_father.add_child(father)
grandfather_of_father.add_child(uncle) # Add uncle as child of paternal grandparents
grandmother_of_father.add_child(uncle)
grandfather_of_father.add_child(aunt) # Add aunt as child of paternal grandparents
grandmother_of_father.add_child(aunt)
grandfather_of_mother.add_child(mother)
grandmother_of_mother.add_child(mother)
family.update([father, mother, uncle, aunt_by_marriage, aunt, uncle_by_marriage])
family.add(father)
if family_size > 3:
mother = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
father.add_spouse(mother)
family.add(mother)
couples.append((father, mother))
if family_size > 6:
grandfather_of_mother.add_child(mother)
grandmother_of_mother.add_child(mother)
if family_size > 8:
# Create father's brother (uncle) and his wife
uncle = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
aunt_by_marriage = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
uncle.add_spouse(aunt_by_marriage)
grandfather_of_father.add_child(uncle) # Add uncle as child of paternal grandparents
grandmother_of_father.add_child(uncle)
family.update([uncle, aunt_by_marriage])
couples.append((uncle, aunt_by_marriage))
if family_size > 10:
# Create father's sister (aunt) and her husband
aunt = Person(get_name(Gender.FEMALE), Gender.FEMALE, next(id_counter))
uncle_by_marriage = Person(get_name(Gender.MALE), Gender.MALE, next(id_counter))
aunt.add_spouse(uncle_by_marriage)
grandfather_of_father.add_child(aunt) # Add aunt as child of paternal grandparents
grandmother_of_father.add_child(aunt)
family.update([aunt, uncle_by_marriage])
couples.append((aunt, uncle_by_marriage))
# 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)