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"]