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

lint

Browse source
This commit is contained in:
Zafir Stojanovski 2025-02-18 14:24:07 +01:00
parent fd5c47d634
commit 6b0ea7a14c
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
reasoning_gym/algorithmic/palindrome_partitioning.py
View file

@ -46,7 +46,9 @@ class PalindromePartitioningConfig:
assert 1 <= self.min_string_len, "Minimum string length must be at least 1"
assert self.min_string_len <= self.max_string_len, "Minimum string length must be less than or equal to maximum"
assert 1 <= self.max_substring_palindome_len, "Maximum substring palindrome length must be at least 1"
assert self.max_substring_palindome_len <= self.max_string_len, "Maximum substring palindrome length must be less than or equal to maximum string length"
assert (
self.max_substring_palindome_len <= self.max_string_len
), "Maximum substring palindrome length must be less than or equal to maximum string length"
class PalindromePartitioningDataset(ProceduralDataset):
@ -119,10 +121,7 @@ class PalindromePartitioningDataset(ProceduralDataset):
output = ""
while len(output) < size:
palindrome_len = rng.randint(
1,
min(self.config.max_substring_palindome_len, size - len(output))
)
palindrome_len = rng.randint(1, min(self.config.max_substring_palindome_len, size - len(output)))
substring = "".join(self._generate_palindrome_letters(rng, palindrome_len))
output += substring