diff --git a/reasoning_gym/logic/__init__.py b/reasoning_gym/logic/__init__.py index 99c2d56f..db976fba 100644 --- a/reasoning_gym/logic/__init__.py +++ b/reasoning_gym/logic/__init__.py @@ -4,21 +4,19 @@ Logic tasks for training reasoning capabilities. from .aiw import AliceInWonderlandConfig, AliceInWonderlandDataset from .circuit_logic import CircuitLogicConfig, CircuitLogicDataset +from .propositional_logic import PropositionalLogicConfig, PropositionalLogicDataset from .self_reference import SelfReferenceConfig, SelfReferenceDataset -from .syllogisms import SyllogismConfig, SyllogismDataset, Term +from .syllogisms import SyllogismConfig, SyllogismDataset from .zebra_puzzles import ZebraConfig, ZebraDataset -# from .propositional_logic import PropositionalLogicConfig, PropositionalLogicDataset - __all__ = [ "AliceInWonderlandConfig", "AliceInWonderlandDataset", - # "PropositionalLogicConfig", - # "PropositionalLogicDataset", + "PropositionalLogicConfig", + "PropositionalLogicDataset", "SyllogismConfig", "SyllogismDataset", "syllogism_dataset", - "Term", "ZebraConfig", "ZebraDataset", "SelfReference", diff --git a/reasoning_gym/logic/propositional_logic.py b/reasoning_gym/logic/propositional_logic.py index 5f895d3b..dec8a5a7 100644 --- a/reasoning_gym/logic/propositional_logic.py +++ b/reasoning_gym/logic/propositional_logic.py @@ -294,7 +294,7 @@ class PropositionalLogicDataset(ProceduralDataset): else: return 1 + self._measure_complexity(expression.left) + self._measure_complexity(expression.right) - def score_answer(self, answer: str | None, entry: Dict[str, Any]) -> float: + def score_answer(self, answer: str | None, entry: dict[str, Any]) -> float: """Robust scoring implementation for propositional logic answers""" if not answer: return 0.0