added indexing for simulated obsdata being split

beast/fitting/fit.py

@@ -902,6 +902,11 @@ def IAU_names_and_extra_info(obsdata, surveyname="PHAT", extraInfo=False):
  obsfiltname = obsdata.filter_aliases[filtername]
  r[filtername] = (obsdata.data[obsfiltname] * obsdata.vega_flux[k]).astype(float)
 
+ # if running a simulation, propagate beast_idx numbers to resort to input obs file
+ if "beast_idx" in list(obsdata.data.keys()):
+ r["beast_idx"] = obsdata.data["beast_idx"]
+
+
  return r
 
 

beast/tools/split_simulated_catalog.py

@@ -92,6 +92,10 @@ def split_simulated_catalog(
  """
  cat = Table.read(catfile)
 
+ # add index column to help sort merged files later
+ cat["beast_idx"] = np.arange(len(cat))
+
+
  # write out sub-files, if chosen
  if (n_per_file is not None) or (min_n_subfile is not None):