From 321b2a88e082711589d967a2ff3de31e68f2935c Mon Sep 17 00:00:00 2001
From: Andreas Koepf <andreas.koepf@provisio.com>
Date: Thu, 13 Feb 2025 17:50:17 +0100
Subject: [PATCH] feat: Add PowerFunctionConfig and PowerFunctionDataset to
 arithmetic module exports

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

diff --git a/reasoning_gym/arithmetic/__init__.py b/reasoning_gym/arithmetic/__init__.py
index cc94bf56..4c10e3b2 100644
--- a/reasoning_gym/arithmetic/__init__.py
+++ b/reasoning_gym/arithmetic/__init__.py
@@ -31,6 +31,8 @@ __all__ = [
     "LCMDataset",
     "LegCountingConfig",
     "LegCountingDataset",
+    "PowerFunctionConfig",
+    "PowerFunctionDataset",
     "PrimeFactorizationConfig",
     "PrimeFactorizationDataset",
     "GSMSymbolicDatasetConfig",