From 53f7a9238c87bdb023c8754708a219eb49844efc Mon Sep 17 00:00:00 2001
From: "Andreas Koepf (aider)" <andreas.koepf@aleph-alpha.com>
Date: Fri, 24 Jan 2025 17:25:35 +0100
Subject: [PATCH] refactor: Use StrEnum and lowercase values for Gender and
 Relationship enums

---
 reasoning_gym/graphs/family_relationships.py | 26 ++++++++++----------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/reasoning_gym/graphs/family_relationships.py b/reasoning_gym/graphs/family_relationships.py
index 614a2128..db30d69f 100644
--- a/reasoning_gym/graphs/family_relationships.py
+++ b/reasoning_gym/graphs/family_relationships.py
@@ -1,28 +1,28 @@
 import random
 from dataclasses import dataclass
-from enum import Enum
+from enum import StrEnum
 from itertools import count
 from typing import Dict, List, Optional, Set, Tuple
 
 from ..dataset import ProceduralDataset
 
 
-class Gender(Enum):
+class Gender(StrEnum):
     MALE = "male"
     FEMALE = "female"
 
 
-class Relationship(Enum):
-    MOTHER = "Mother"
-    FATHER = "Father"
-    SISTER = "Sister"
-    BROTHER = "Brother"
-    DAUGHTER = "Daughter"
-    SON = "Son"
-    WIFE = "Wife"
-    HUSBAND = "Husband"
-    GRANDMOTHER = "Grandmother"
-    GRANDFATHER = "Grandfather"
+class Relationship(StrEnum):
+    MOTHER = "mother"
+    FATHER = "father"
+    SISTER = "sister"
+    BROTHER = "brother"
+    DAUGHTER = "daughter"
+    SON = "son"
+    WIFE = "wife"
+    HUSBAND = "husband"
+    GRANDMOTHER = "grandmother"
+    GRANDFATHER = "grandfather"
 
 
 @dataclass