from logic.variants import (lin2015_variant,
                            logs_normal_ig_variant,
                            normal_ig_variant,
                            log_matrix_normal_variant,
                            log_mv_normal_variant,
                            log_mv_normal_double_chol_variant,
                            mv_normal_variant,
                            mv_normal_double_chol_variant,
                            matrix_normal_variant)
from logic.runners import run_variant
from data import dataset

if __name__ == '__main__':
    # run_variant(lin2015_variant, dataset, "lin")
    # run_variant(logs_normal_ig_variant, dataset, "logs_nig")
    # run_variant(normal_ig_variant, dataset, "nig")
    run_variant(log_matrix_normal_variant, {"a1": dataset["a1"]}, "log_matrix_rev_chol", mcmc=True)
    # run_variant(log_mv_normal_variant, dataset, "log_mv_reversed", mcmc=True)
    # run_variant(log_mv_normal_double_chol_variant, {"a4": dataset["a4"],
    #                                                 "a5": dataset["a5"],
    #                                                 "a6": dataset["a6"]}, "log_mv_chol", mcmc=True)