open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove unmodified dunder methods

Browse source
This commit is contained in:
Zafir Stojanovski 2025-02-06 22:56:11 +01:00
parent 071c22a809
commit b5a820a8d9
3 changed files with 0 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

14
reasoning_gym/games/n_queens.py
View file

@ -50,20 +50,6 @@ class NQueensDataset(ProceduralDataset):
super().__init__(config=config, seed=config.seed, size=config.size)
self._solutions = self._get_all_solutions(config.n)
def __len__(self) -> int:
return self.config.size
def __iter__(self):
self._current_idx = 0
return self
def __next__(self):
if self._current_idx >= self.config.size:
raise StopIteration
item = self[self._current_idx]
self._current_idx += 1
return item
def _get_all_solutions(self, n: int) -> List[List[List[str]]]:
"""Get all solutions for the N Queens puzzle"""

14
reasoning_gym/graphs/course_schedule.py
View file

@ -54,20 +54,6 @@ class CourseScheduleDataset(ProceduralDataset):
def __init__(self, config: CourseScheduleConfig):
super().__init__(config=config, seed=config.seed, size=config.size)
def __len__(self) -> int:
return self.config.size
def __iter__(self):
self._current_idx = 0
return self
def __next__(self):
if self._current_idx >= self.config.size:
raise StopIteration
item = self[self._current_idx]
self._current_idx += 1
return item
def _can_finish(self, num_courses: int, prerequisites: List[List[int]]) -> bool:
adj = defaultdict(list)
for course, prereq in prerequisites:

14
reasoning_gym/graphs/largest_island.py
View file

@ -55,20 +55,6 @@ class LargestIslandDataset(ProceduralDataset):
def __init__(self, config: LargestIslandConfig):
super().__init__(config=config, seed=config.seed, size=config.size)
def __len__(self) -> int:
return self.config.size
def __iter__(self):
self._current_idx = 0
return self
def __next__(self):
if self._current_idx >= self.config.size:
raise StopIteration
item = self[self._current_idx]
self._current_idx += 1
return item
def _is_valid_cell(self, r: int, c: int) -> bool:
return 0 <= r < self.config.rows and 0 <= c < self.config.cols