From 184d5b42700a41096b0a5f2dbed738ffadabd33a Mon Sep 17 00:00:00 2001
From: joesharratt1229 <joesharratt1229@gmail.com>
Date: Sat, 8 Feb 2025 09:45:49 +0000
Subject: [PATCH] registered rearc

---
 reasoning_gym/__init__.py     | 3 ++-
 reasoning_gym/arc/__init__.py | 3 +++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/reasoning_gym/__init__.py b/reasoning_gym/__init__.py
index ecca7f3f..48be1fa9 100644
--- a/reasoning_gym/__init__.py
+++ b/reasoning_gym/__init__.py
@@ -2,11 +2,12 @@
 Reasoning Gym - A library of procedural dataset generators for training reasoning models
 """
 
-from . import algebra, algorithmic, arithmetic, code, cognition, data, games, geometry, graphs, logic
+from . import algebra, algorithmic, arc, arithmetic, code, cognition, data, games, geometry, graphs, logic
 from .factory import create_dataset, register_dataset
 
 __version__ = "0.1.5"
 __all__ = [
+    "arc",
     "algebra",
     "algorithmic",
     "arithmetic",
diff --git a/reasoning_gym/arc/__init__.py b/reasoning_gym/arc/__init__.py
index e69de29b..e1ca90b8 100644
--- a/reasoning_gym/arc/__init__.py
+++ b/reasoning_gym/arc/__init__.py
@@ -0,0 +1,3 @@
+from .rearc import ReArcConfig, ReArcDataset
+
+__all__ = ["ReArcDataset", "ReArcConfig"]