diff options
Diffstat (limited to 'utils/mcmc.py')
| -rw-r--r-- | utils/mcmc.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/utils/mcmc.py b/utils/mcmc.py index c712c3a..97b77dd 100644 --- a/utils/mcmc.py +++ b/utils/mcmc.py @@ -9,7 +9,6 @@ Useful functions to use an MCMC for the BSM flavour ratio analysis from __future__ import absolute_import, division -import argparse from functools import partial import emcee @@ -72,6 +71,14 @@ def mcmc_argparse(parser): type=partial(enum_parse, c=MCMCSeedType), choices=MCMCSeedType, help='Type of distrbution to make the initial MCMC seed' ) + parser.add_argument( + '--plot-angles', type=misc_utils.parse_bool, default='True', + help='Plot MCMC triangle in the angles space' + ) + parser.add_argument( + '--plot-elements', type=misc_utils.parse_bool, default='False', + help='Plot MCMC triangle in the mixing elements space' + ) def flat_seed(paramset, nwalkers): |