From e1d04f8a2f8cf55e9faee0e50e3b17f12d29ddac Mon Sep 17 00:00:00 2001
From: "Andreas Koepf (aider)" <andreas.koepf@provisio.com>
Date: Thu, 23 Jan 2025 13:31:30 +0100
Subject: [PATCH] feat: Add cognition module with initial reasoning tasks
 namespace

---
 python                    | 9 +++++++++
 reasoning_gym/__init__.py | 3 ++-
 2 files changed, 11 insertions(+), 1 deletion(-)
 create mode 100644 python

diff --git a/python b/python
new file mode 100644
index 00000000..289a79d6
--- /dev/null
+++ b/python
@@ -0,0 +1,9 @@
+"""
+Cognition tasks for training reasoning capabilities:
+- Pattern recognition
+- Sequence completion
+- Logical reasoning
+- Working memory
+"""
+
+__all__ = []
diff --git a/reasoning_gym/__init__.py b/reasoning_gym/__init__.py
index c1b93a3a..aca8e226 100644
--- a/reasoning_gym/__init__.py
+++ b/reasoning_gym/__init__.py
@@ -3,6 +3,7 @@ Reasoning Gym - A library of procedural dataset generators for training reasonin
 """
 
 from . import arithmetic
+from . import cognition
 
 __version__ = "0.1.0"
-__all__ = ["arithmetic"]
+__all__ = ["arithmetic", "cognition"]