reasoning-gym/reasoning_gym/logic/__init__.py
vncntt 5f01049607
Add KnightsKnavesDataset (knights_knaves)
Adapted code from https://github.com/AlphaPav/mem-kk-logic/blob/main/data_prep/lib_kk.py

---------

Co-authored-by: Andreas Koepf (aider) <andreas.koepf@provisio.com>
2025-02-25 20:15:38 +01:00

30 lines
966 B
Python

"""
Logic tasks for training reasoning capabilities.
"""
from .aiw import AliceInWonderlandConfig, AliceInWonderlandDataset
from .circuit_logic import CircuitLogicConfig, CircuitLogicDataset
from .knights_knaves import KnightsKnavesConfig, KnightsKnavesDataset
from .propositional_logic import PropositionalLogicConfig, PropositionalLogicDataset
from .self_reference import SelfReferenceConfig, SelfReferenceDataset
from .syllogisms import SyllogismConfig, SyllogismDataset
from .zebra_puzzles import ZebraConfig, ZebraDataset
__all__ = [
"AliceInWonderlandConfig",
"AliceInWonderlandDataset",
"PropositionalLogicConfig",
"PropositionalLogicDataset",
"SyllogismConfig",
"SyllogismDataset",
"syllogism_dataset",
"ZebraConfig",
"ZebraDataset",
"SelfReference",
"SelfReferenceConfig",
"SelfReferenceDataset",
"CircuitLogicConfig",
"CircuitLogicDataset",
"KnightsKnavesConfig",
"KnightsKnavesDataset",
]