Example #1
0
def prepare_data_script():
    warn_deprecation()
    logger_setup()
    if "CONTAINED" not in os.environ:
        log.error("This command should only be run within a container. "
                  "Run 'cobaya-install' instead.")
        raise HandledException
    parser = argparse.ArgumentParser(
        description=
        "Cobaya's installation tool for the data needed by a container.")
    parser.add_argument(
        "-f",
        "--force",
        action="store_true",
        default=False,
        help="Force re-installation of apparently installed modules.")
    arguments = parser.parse_args()
    try:
        info = load_input(requirements_file_path)
    except IOError:
        log.error(
            "Cannot find the requirements file. This should not be happening.")
        raise HandledException
    install(info,
            path=_modules_path,
            force=arguments.force,
            **{
                _code: False,
                _data: True
            })
Example #2
0
    def setUp(self):
        from cobaya.install import install

        install(
            {"likelihood": {"planck_2020_lollipop.lowlB": None}},
            path=packages_path,
            skip_global=True,
        )
Example #3
0
    def setUp(self):
        from cobaya.install import install

        install({"likelihood": {
            "mflike.MFLike": None
        }},
                path=packages_path,
                skip_global=True)
Example #4
0
    def setUp(self):
        from cobaya.install import install

        for mode in chi2s.keys():
            install(
                {"likelihood": {"planck_2020_hillipop.{}".format(mode): None}},
                path=packages_path,
                skip_global=True,
            )
Example #5
0
            "planck_2015_lowTEB": None,
            "planck_2015_plikHM_TTTEEE": None,
            "planck_2015_lensing": None,
            "planck_2015_lensing_cmblikes": None,
            "bicep_keck_2015": None,
            "sn_pantheon": None,
            "sn_jla": None,
            "sn_jla_lite": None,
            "sdss_dr12_consensus_bao": None,
            "sdss_dr12_consensus_full_shape": None,
            "sdss_dr12_consensus_final": None,
            "des_y1_joint": None
        }
    }
    # Ignore Planck clik in Python 3 or GCC > 5
    if sys.version_info.major >= 3 or not config.gcc_version_atleast('6.0'):
        popliks = [
            lik for lik in info_install[_likelihood]
            if lik.startswith("planck_2015") and not lik.endswith("cmblikes")
        ]
        for lik in popliks:
            info_install[_likelihood].pop(lik)
    path = sys.argv[1]
    import os

    if not os.path.exists(path):
        os.makedirs(path)
    from cobaya.install import install

    install(info_install, path=path, no_progress_bars=True)
Example #6
0
    def setUp(self):
        from cobaya.install import install

        install({"likelihood": {"spt_hiell_2020.TT": None}}, path=packages_path)
Example #7
0
    def setUp(self):
        from cobaya.install import install

        install({"likelihood": {"sptpol_2017.TEEE": None}}, path=packages_path)
Example #8
0
def get_demo_lensing_model(theory):
    if theory == "camb":
        info_yaml = r"""
        likelihood:
            soliket.LensingLikelihood:
                stop_at_error: True

        theory:
            camb:
                extra_args:
                    lens_potential_accuracy: 1

        params:
            ns:
                prior:
                  min: 0.8
                  max: 1.2
            H0:
                prior:
                  min: 40
                  max: 100        
        """
    elif theory == "classy":
        info_yaml = r"""
        likelihood:
            soliket.LensingLikelihood:
                stop_at_error: True

        theory:
            classy:
                extra_args:
                    output: lCl, tCl
                path: global

        params:
            n_s:
                prior:
                  min: 0.8
                  max: 1.2
            H0:
                prior:
                  min: 40
                  max: 100        

        """

    info = yaml_load(info_yaml)

    from cobaya.install import install
    install(info, path=packages_path, skip_global=True)

    test_point = {}
    for par, pdict in info["params"].items():
        if not isinstance(pdict, dict):
            continue

        if "ref" in pdict:
            try:
                value = float(pdict["ref"])
            except TypeError:
                value = (pdict["ref"]["min"] + pdict["ref"]["max"]) / 2
            test_point[par] = value
        elif "prior" in pdict:
            value = (pdict["prior"]["min"] + pdict["prior"]["max"]) / 2
            test_point[par] = value

    model = get_model(info)
    return model, test_point
Example #9
0
 def setUp(self):
     from cobaya.install import install
     install({"likelihood": {"mflike.MFLike": None}}, path=modules_path)