enable some CLI options

paynt/cli.py

@@ -75,6 +75,10 @@ def setup_logger(log_path = None):
 
 @click.option("--incomplete-search", is_flag=True, default=False,
     help="use incomplete search during synthesis")
+@click.option("--disable-expected-visits", is_flag=True, default=False,
+    help="do not compute expected visits for the splitting heuristic")
+@click.option("--split-mdp-family-pessimistically", is_flag=True, default=False,
+    help="when solving families of MDPs, split undecided families without using game solution")
 
 @click.option("--fsc-synthesis", is_flag=True, default=False,
     help="enable incremental synthesis of FSCs for a POMDP")
@@ -129,7 +133,8 @@ def paynt_run(
     project, sketch, props, relative_error, discount_factor,
     export,
     method,
-    incomplete_search,
+    incomplete_search, disable_expected_visits,
+    split_mdp_family_pessimistically,
     fsc_synthesis, pomdp_memory_size, posterior_aware,
     fsc_export_result,
     storm_pomdp, iterative_storm, get_storm_result, storm_options, prune_storm,
@@ -143,11 +148,14 @@ def paynt_run(
 
     # set CLI parameters
     Synthesizer.incomplete_search = incomplete_search
+    paynt.quotient.quotient.QuotientContainer.compute_expected_visits = not disable_expected_visits
+    paynt.synthesizer.policy_tree.SynthesizerPolicyTree.use_optimistic_splitting = not split_mdp_family_pessimistically
     SynthesizerCEGIS.conflict_generator_type = ce_generator
     paynt.quotient.quotient_pomdp.POMDPQuotientContainer.initial_memory_size = pomdp_memory_size
     paynt.quotient.quotient_pomdp.POMDPQuotientContainer.export_optimal_result = fsc_export_result
     paynt.quotient.quotient_pomdp.POMDPQuotientContainer.posterior_aware = posterior_aware
 
+
     # join paths of input files
     sketch_path = os.path.join(project, sketch)
     properties_path = os.path.join(project, props)