Merge pull request #594 from aiplan4eu/fix-implies-problem-kind

Fix IMPLY ProblemKind
aiplan4eu · Apr 12, 2024 · eca4ee0 · eca4ee0
2 parents 1a540af + 722a58d
commit eca4ee0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/unified_planning/model/multi_agent/ma_problem.py b/unified_planning/model/multi_agent/ma_problem.py
@@ -367,7 +367,7 @@ def _update_problem_kind_condition(self, exp: "up.model.fnode.FNode"):
             self._kind.set_conditions_kind("EQUALITIES")
         if OperatorKind.NOT in ops:
             self._kind.set_conditions_kind("NEGATIVE_CONDITIONS")
-        if OperatorKind.OR in ops:
+        if OperatorKind.OR in ops or OperatorKind.IMPLIES in ops:
             self._kind.set_conditions_kind("DISJUNCTIVE_CONDITIONS")
         if OperatorKind.EXISTS in ops:
             self._kind.set_conditions_kind("EXISTENTIAL_CONDITIONS")

diff --git a/unified_planning/model/problem.py b/unified_planning/model/problem.py
@@ -866,7 +866,7 @@ def update_problem_kind_expression(
             self.kind.set_conditions_kind("EQUALITIES")
         if OperatorKind.NOT in ops:
             self.kind.set_conditions_kind("NEGATIVE_CONDITIONS")
-        if OperatorKind.OR in ops:
+        if OperatorKind.OR in ops or OperatorKind.IMPLIES in ops:
             self.kind.set_conditions_kind("DISJUNCTIVE_CONDITIONS")
         if OperatorKind.EXISTS in ops:
             self.kind.set_conditions_kind("EXISTENTIAL_CONDITIONS")