fix usage of new Coloring for POMDP families

randriu · Dec 10, 2023 · 1024152 · 1024152
1 parent f9d0a24
commit 1024152
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/paynt/quotient/pomdp_family.py b/paynt/quotient/pomdp_family.py
@@ -214,7 +214,7 @@ def build_dtmc_sketch(self, fsc):
             else:
                 hole_options = [(hole,option) for hole,option in choice_to_hole_assignment[choice]]
             product_choice_to_hole_options.append(hole_options)
-        product_coloring = stormpy.synthesis.Coloring(product_family.family, product.nondeterministic_choice_indices, choice_to_hole_options)
+        product_coloring = stormpy.synthesis.Coloring(product_family.family, product.nondeterministic_choice_indices, product_choice_to_hole_options)
 
         # handle specification
         product_specification = self.specification.copy()

diff --git a/paynt_pomdp_sketch.py b/paynt_pomdp_sketch.py
@@ -67,8 +67,7 @@ def investigate_hole_assignment(pomdp_sketch, hole_assignment):
 print("design space size: {} members".format(pomdp_sketch.design_space.size))
 
 # fix some hole options
-hole_options = [[hole.options[0]] for hole in pomdp_sketch.design_space]
-hole_assignment = pomdp_sketch.design_space.assume_options_copy(hole_options)
+hole_assignment = pomdp_sketch.design_space.pick_any()
 
 # investigate this hole assignment and return an FSC
 fsc = investigate_hole_assignment(pomdp_sketch, hole_assignment)