add --universe argument to dm-search

1 files changed, 2 insertions, 0 deletions

diff --git a/utils/chi2 b/utils/chi2
index ac4093a..f32208a 100755
--- a/utils/chi2
+++ b/utils/chi2
@@ -418,6 +418,8 @@ def do_fit(data,muon,data_mc,weights,atmo_scale_factor,muon_scale_factor,bins,st
 
     universe = np.argmin(nlls)
 
+    print("universe = ", universe)
+
     if refit:
         data_mc_with_weights = pd.merge(data_mc,weights[universe],how='left',on=['run','unique_id'])
         data_mc_with_weights.weight = data_mc_with_weights.weight.fillna(1.0)