From b8c69aa89f7a33a1a4c513b394ce543271d1b8a8 Mon Sep 17 00:00:00 2001
From: "Andreas Koepf (aider)" <andreas.koepf@aleph-alpha.com>
Date: Thu, 23 Jan 2025 22:22:12 +0100
Subject: [PATCH] feat: Add leg counting imports and exports to arithmetic
 package

---
 reasoning_gym/arithmetic/__init__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/reasoning_gym/arithmetic/__init__.py b/reasoning_gym/arithmetic/__init__.py
index ad31381e..05d42b6d 100644
--- a/reasoning_gym/arithmetic/__init__.py
+++ b/reasoning_gym/arithmetic/__init__.py
@@ -1,9 +1,13 @@
 from .chain_sum import ChainSum, ChainSumConfig
 from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig
+from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_dataset
 
 __all__ = [
     "ChainSum",
     "ChainSumConfig",
     "ArithmeticDataset",
-    "ArithmeticDatasetConfig"
+    "ArithmeticDatasetConfig",
+    "LegCountingConfig",
+    "LegCountingDataset",
+    "leg_counting_dataset"
 ]