remove triplicate allocation

GarethCabournDavies · Dec 12, 2024 · c8da9ef · c8da9ef
1 parent dc7eab7
commit c8da9ef
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/bin/all_sky_search/pycbc_fit_sngls_over_multiparam b/bin/all_sky_search/pycbc_fit_sngls_over_multiparam
@@ -467,33 +467,24 @@ elif numpy.isfinite(_smooth_cut[args.smoothing_method]):
     nabove_sort =  nabove[par_sort]
     invalphan_sort = invalphan[par_sort]
     ntotal_sort = ntotal[par_sort]
-    # Preallocate memory for *param_vals[0]-sorted* smoothing results
-    nabove_smoothed = numpy.zeros(num_templates)
-    alpha_smoothed = numpy.zeros(num_templates)
-    ntotal_smoothed = numpy.zeros(num_templates)
     for i in rang:
         report_percentage(i, num_templates)
         slc = slices[i]
         d = dist(i, slc, parvals, args.smoothing_width)
 
-        smoothed_tuple = smooth(
+        smoothed_vals[i,:] = smooth(
             nabove_sort[slc],
             invalphan_sort[slc],
             ntotal_sort[slc],
             d,
             args.smoothing_method,
             **kwarg_dict
         )
-        nabove_smoothed[i] = smoothed_tuple[0]
-        alpha_smoothed[i] = smoothed_tuple[1]
-        ntotal_smoothed[i] = smoothed_tuple[2]
 
     # Undo the sorts
     unsort = numpy.argsort(par_sort)
     parvals = [p[unsort] for p in parvals]
-    smoothed_vals[:,0] = nabove_smoothed[unsort]
-    smoothed_vals[:,1] = alpha_smoothed[unsort]
-    smoothed_vals[:,2] = ntotal_smoothed[unsort]
+    smoothed_vals = smoothed_vals[unsort,:]
 
 else:
     logging.info("Smoothing ...")