Add indexes to df grouping to preserve integer ordering.

lsst · Nov 2, 2023 · eb5f2fd · eb5f2fd
1 parent 850a35b
commit eb5f2fd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/schedview/compute/scheduler.py b/schedview/compute/scheduler.py
@@ -399,6 +399,8 @@ def make_survey_row(survey_bfs):
         survey_row = pd.Series({"reward": reward, "infeasible": infeasible})
         return survey_row
 
-    survey_df = summary_df.groupby(["tier", "survey_name_with_id", "survey_url"]).apply(make_survey_row)
+    survey_df = summary_df.groupby(
+        ["list_index", "survey_index", "survey_name_with_id", "survey_url", "tier"]
+    ).apply(make_survey_row)
 
     return survey_df["reward"].reset_index()