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

fix: Correctly handle empty name lists in FamilyRelationshipsConfig

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 16:59:18 +01:00
parent 6a81b7a8f7
commit 630c6503c0
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
reasoning_gym/graphs/family_relationships.py
View file

@ -69,14 +69,19 @@ class FamilyRelationshipsConfig:
def __post_init__(self):
# Default name lists if none provided
self.male_names = self.male_names or [
default_male_names = [
"James", "John", "Robert", "Michael", "William", "David", "Richard",
"Joseph", "Thomas", "Charles", "Peter", "Daniel", "Matthew"
]
self.female_names = self.female_names or [
default_female_names = [
"Mary", "Patricia", "Jennifer", "Linda", "Elizabeth", "Barbara", "Susan",
"Jessica", "Sarah", "Karen", "Emma", "Lisa", "Anna"
]
if self.male_names is None:
self.male_names = default_male_names
if self.female_names is None:
self.female_names = default_female_names
def validate(self):
"""Validate configuration parameters"""