relaxer add structure

potassco · Apr 11, 2023 · f7908fe · f7908fe
1 parent 4fea3f1
commit f7908fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/backend/src/viasp/api.py b/backend/src/viasp/api.py
@@ -20,7 +20,7 @@
 
 from .shared.defaults import STDIN_TMP_STORAGE_PATH
 from .shared.io import clingo_symbols_to_stable_model
-from .wrapper import ShowConnector
+from .wrapper import ShowConnector, Control as viaspControl
 
 __all__ = [
     "load_program_file",
@@ -279,7 +279,7 @@ def get_relaxed_program(*args, **kwargs) -> str:
     connector = _get_connector(**kwargs)
     return connector.get_relaxed_program(head_name, collect_variables)
 
-def relax_constraints(*args, **kwargs):
+def relax_constraints(*args, **kwargs) -> viaspControl:
     r"""
     Relax constraints in the marked models. Returns 
     a new viaspControl object with the relaxed program loaded

diff --git a/backend/src/viasp/wrapper.py b/backend/src/viasp/wrapper.py
@@ -58,6 +58,8 @@ def get_relaxed_program(self,  head_name:str = "unsat", collect_variables:bool =
         :param collect_variables: ``bool``
             default=True (collect variables from body as a tuple in the head literal)
         """
+        self._database.set_target_stable_model(self._marked)
+        self._database._reconstruct()
         kwargs = {"head_name": head_name, "collect_variables": collect_variables}
         return self._database.relax_constraints(**kwargs)