knights_knaves

NOTICE.txt

@@ -60,6 +60,18 @@ or served as inspiration for parts of our codebase:
 - License: MIT
 - Copyright (c) 2024 AlphaPav
 - Note: Adapted for K&K problem generation
+- Citation:
+```
+@article{xie2024memorization,
+      title={On Memorization of Large Language Models in Logical Reasoning},
+      author={Chulin Xie and Yangsibo Huang and Chiyuan Zhang and Da Yu and Xinyun Chen and Bill Yuchen Lin and Bo Li and Badih Ghazi and Ravi Kumar},
+      year={2024},
+      eprint={2410.23123},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2410.23123},
+}
+```
 
 We are deeply grateful to all authors of these projects for making their work available
 under open source licenses, enabling us to build upon their contributions.

reasoning_gym/arithmetic/decimal_chain_sum.py

@@ -3,7 +3,6 @@
 from decimal import Decimal
 from typing import Any, Optional
 
-from ..coaching import AttributeType, BaseCurriculum, RangeAttributeDefinition
 from ..factory import ProceduralDataset, register_dataset
 
 

reasoning_gym/logic/__init__.py

@@ -4,6 +4,7 @@
 
 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
@@ -24,4 +25,6 @@
     "SelfReferenceDataset",
     "CircuitLogicConfig",
     "CircuitLogicDataset",
+    "KnightsKnavesConfig",
+    "KnightsKnavesDataset",
 ]