fix(interpreted functions): functions in durations should now work

aiplan4eu · Sep 16, 2024 · 7520f30 · 7520f30
1 parent aad6735
commit 7520f30
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/unified_planning/model/problem.py b/unified_planning/model/problem.py
@@ -935,6 +935,7 @@ def update_action_parameter(self, param: "up.model.Parameter"):
 
     def update_action_duration(self, duration: "up.model.DurationInterval"):
         lower, upper = duration.lower, duration.upper
+
         for dur_bound in (lower, upper):
             if dur_bound.type.is_int_type():
                 self.kind.set_expression_duration("INT_TYPE_DURATIONS")
@@ -947,6 +948,9 @@ def update_action_duration(self, duration: "up.model.DurationInterval"):
         free_vars = self.environment.free_vars_extractor.get(
             lower
         ) | self.environment.free_vars_extractor.get(upper)
+        ops = self.operators_extractor.get(lower) | self.operators_extractor.get(upper)
+        if OperatorKind.INTERPRETED_FUNCTION_EXP in ops:
+            self.kind.set_expression_duration("INTERPRETED_FUNCTIONS_IN_DURATIONS")
 
         if len(free_vars) > 0:
             only_static = True