Merge branch 'randriu:master' into new-master

randriu · Jan 9, 2024 · cf48ac8 · cf48ac8
2 parents f99ca87 + 9d692eb
commit cf48ac8
Show file tree

Hide file tree

Showing 5 changed files with 275 additions and 780 deletions.
diff --git a/models/mdp/sketches/dodge/8-mod2-stagger/sketch.props b/models/mdp/sketches/dodge/8-mod2-stagger/sketch.props
diff --git a/models/mdp/sketches/dodge/8-mod2-stagger/sketch.templ b/models/mdp/sketches/dodge/8-mod2-stagger/sketch.templ
diff --git a/paynt/quotient/mdp_family.py b/paynt/quotient/mdp_family.py
@@ -99,6 +99,7 @@ def policy_to_state_valuation_actions(self, policy):
         Create a representation for a policy that associated action labels with state valuations. States with only
         one available action are omitted.
         '''
+        policy,_ = policy
         sv = self.quotient_mdp.state_valuations
         state_valuation_to_action = []
         for state,action in enumerate(policy):
@@ -166,6 +167,8 @@ def fix_and_apply_policy_to_family(self, family, policy):
         for state in mdp.quotient_state_map:
             policy_fixed[state] = policy[state]
 
+        mask = [state for state,action in enumerate(policy_fixed) if action is not None]
+        policy_fixed = (policy_fixed,mask)
         return policy_fixed,mdp