[aiw] remove output format enum

open-thought · Feb 2, 2025 · f8696d6 · f8696d6
1 parent 3d42e84
commit f8696d6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git a/reasoning_gym/logic/aiw.py b/reasoning_gym/logic/aiw.py
@@ -14,13 +14,6 @@ class TaskType(Enum):
     COLLEAGUES = "colleagues"  # Added colleagues task
 
 
-class OutputFormat(Enum):
-    """Defines the output format for the generated questions."""
-    PLAIN = "plain"
-    RESTRICTED = "restricted"
-    THINKING = "thinking"
-
-
 @dataclass
 class AliceInWonderlandConfig:
     """Configuration options for the Alice in Wonderland dataset.

diff --git a/tests/test_aiw.py b/tests/test_aiw.py
@@ -1,6 +1,6 @@
 import pytest
 
-from reasoning_gym.logic.aiw import AliceInWonderlandConfig, AliceInWonderlandDataset, TaskType, OutputFormat
+from reasoning_gym.logic.aiw import AliceInWonderlandConfig, AliceInWonderlandDataset, TaskType
 
 
 def test_aiw_config_validation():