From 4d9ea46c8528ffcecdea78caa3eda585e1d3bb0a Mon Sep 17 00:00:00 2001
From: joesharratt1229 <joesharratt1229@gmail.com>
Date: Sat, 8 Feb 2025 10:21:04 +0000
Subject: [PATCH] added validation

---
 reasoning_gym/arc/rearc.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/reasoning_gym/arc/rearc.py b/reasoning_gym/arc/rearc.py
index 029da94e..88762f51 100644
--- a/reasoning_gym/arc/rearc.py
+++ b/reasoning_gym/arc/rearc.py
@@ -58,6 +58,10 @@ class ReArcConfig:
     seed: Optional[int] = None
     size: int = 500
 
+    def validate(self):
+        assert self.diff_lb < self.diff_ub, "diff_lb must be < diff_ub."
+        assert self.size > 0, "Size of dataset must be positive."
+
 
 class ReArcDataset(ProceduralDataset):
     def __init__(self, config: ReArcConfig):