From 99b9d7b8ab41de3c3239ace50413445f19ef6798 Mon Sep 17 00:00:00 2001 From: Roman Andriushchenko Date: Thu, 7 Dec 2023 16:45:51 +0100 Subject: [PATCH] enable some CLI options --- paynt/cli.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/paynt/cli.py b/paynt/cli.py index 589dfc5e9..e19206bf0 100644 --- a/paynt/cli.py +++ b/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)