From 3046b531a219e259f1773710ec5ffff6c8e52d64 Mon Sep 17 00:00:00 2001 From: Susana Hahn Date: Thu, 8 Aug 2024 19:24:27 -0600 Subject: [PATCH] Added return value to simplify --- src/clingexplaid/mus/core_computer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/clingexplaid/mus/core_computer.py b/src/clingexplaid/mus/core_computer.py index b43de77..38fa344 100644 --- a/src/clingexplaid/mus/core_computer.py +++ b/src/clingexplaid/mus/core_computer.py @@ -75,12 +75,15 @@ def _compute_single_minimal(self, assumptions: Optional[AssumptionSet] = None) - return mus_members - def shrink(self, assumptions: Optional[AssumptionSet] = None) -> None: + def shrink(self, assumptions: Optional[AssumptionSet] = None) -> AssumptionSet: """ This function applies the unsatisfiable subset minimization (`self._compute_single_minimal`) on the assumptions set `assumptions` and stores the resulting MUS inside `self.minimal`. + + Returns the MUS as a set of assumptions. """ self.minimal = self._compute_single_minimal(assumptions=assumptions) + return self.minimal def get_multiple_minimal(self, max_mus: Optional[int] = None) -> Generator[AssumptionSet, None, None]: """