<commit_before>import os

from chainconsumer import ChainConsumer

from dessn.models.e_toplevel.load import plot_quick, plot_debug, plot_separate, load_stan_from_folder

if __name__ == "__main__":
    dir_name = os.path.dirname(__file__)
    std = dir_name + "/stan_output"


    c = ChainConsumer()
    datas = ["snana_dummy", "snana_diff", "snana_diff2", "simple"]
    labels = [r"SNANA $\Omega_m=0.3$", r"SNANA $\Omega_m=0.2$", r"SNANA $\Omega_m=0.4$", "Simple model with mass"]
    for data_source, label in zip(datas, labels):

        chain, posterior, t, p, f, l, w, ow = load_stan_from_folder(dir_name + "/stan_output_%s" % data_source,
                                                                cut=False, merge=True, mod_weight=False)

        c.add_chain(chain, posterior=posterior, name=label)

    c.configure(shade=True)
    c.plot(filename=dir_name + "/colour_all.png", truth=t, parameters=9)

<commit_msg>pep8 fixes. Still thinking about odd population fitting<commit_after>import os

from chainconsumer import ChainConsumer

from dessn.models.e_toplevel.load import plot_quick, plot_debug, plot_separate, load_stan_from_folder

if __name__ == "__main__":
    dir_name = os.path.dirname(__file__)
    std = dir_name + "/stan_output"

    c = ChainConsumer()

    datas = ["snana_dummy", "snana_diff", "snana_diff2", "simple"]
    labels = [r"SNANA $\Omega_m=0.3$", r"SNANA $\Omega_m=0.2$", r"SNANA $\Omega_m=0.4$", "Simple model with mass"]

    for data_source, label in zip(datas, labels):

        chain, posterior, t, p, f, l, w, ow = load_stan_from_folder(dir_name + "/stan_output_%s" % data_source,
                                                                cut=False, merge=True, mod_weight=False)

        c.add_chain(chain, posterior=posterior, name=label)

    c.configure(shade=True)
    c.plot(filename=dir_name + "/colour_all.png", truth=t, parameters=9)


{'fraction_non_alphanumeric': 0.12823218997361477, 'fraction_numerical': 0.008970976253298154, 'mean_word_length': 3.27027027027027, 'pattern_counts': {'":': 2, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 4, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '4934586', 'n_tokens_mistral': 661, 'n_tokens_neox': 637, 'n_words': 141}