From 3ce453eeb5d2d05a746b178dac872d9363f8dfe1 Mon Sep 17 00:00:00 2001 From: David Hendriks <davidhendriks93@gmail.com> Date: Thu, 29 Apr 2021 18:20:35 +0100 Subject: [PATCH] Updated code for M&S --- binarycpython/utils/distribution_functions.py | 2 +- binarycpython/utils/grid.py | 16 +++++++++++++--- 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/binarycpython/utils/distribution_functions.py b/binarycpython/utils/distribution_functions.py index 14e88af30..86f61a857 100644 --- a/binarycpython/utils/distribution_functions.py +++ b/binarycpython/utils/distribution_functions.py @@ -1107,7 +1107,7 @@ def Moe_de_Stefano_2017_multiplicity_fractions(options): for key in result.keys(): result[key] = result[key] / sum_result - print("Multiplicity array: {}".format(str(result))) + # print("Multiplicity array: {}".format(str(result))) # return array reference return result diff --git a/binarycpython/utils/grid.py b/binarycpython/utils/grid.py index 4d2a35edb..0fe333539 100644 --- a/binarycpython/utils/grid.py +++ b/binarycpython/utils/grid.py @@ -38,7 +38,6 @@ import argparse import subprocess import importlib.util - from typing import Union, Any from pathos.helpers import mp as pathos_multiprocess @@ -81,11 +80,14 @@ import binarycpython.utils.moe_distefano_data # get_python_details, # ) -from binarycpython.utils.distribution_functions import Moecache, LOG_LN_CONVERTER +from binarycpython.utils.distribution_functions import ( + Moecache, + LOG_LN_CONVERTER, + Moe_de_Stefano_2017_multiplicity_fractions +) from binarycpython import _binary_c_bindings - class Population: """ Population Object. Contains all the necessary functions to set up, run and process a @@ -3037,6 +3039,14 @@ class Population: triple_values = [multiplicity_dict[key][2] for key in multiplicity_dict.keys()] quad_values = [multiplicity_dict[key][3] for key in multiplicity_dict.keys()] + print("mass = {}".format(list(mass_range))) + print("single_values={}".format(single_values)) + print("binary_values={}".format(binary_values)) + print("triple_values={}".format(triple_values)) + print("quad_values={}".format(quad_values)) + + + plt.plot(mass_range, single_values, label='single') plt.plot(mass_range, binary_values, label='binary') plt.plot(mass_range, triple_values, label='triple') -- GitLab