Removed to record set operation

oislen · Oct 10, 2024 · b009351 · b009351
1 parent 9074224
commit b009351
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/PreProcessData/gen_preaggregate_data.py b/scripts/PreProcessData/gen_preaggregate_data.py
@@ -46,7 +46,7 @@ def gen_preaggregate_data(
         logging.info(f"{stat} ...")
         agg_dict = {col: stat for col in cons.col_options}
         tmp_agg_data = agg_data.groupby(group_cols, as_index=False).agg(agg_dict)
-        pre_agg_data_dict[stat] = tmp_agg_data.to_dict(orient='records')
+        pre_agg_data_dict[stat] = tmp_agg_data
     # if the output
     if preaggregate_data_fpath != None:
         if os.path.exists(preaggregate_data_fpath):