Skip to content

Commit

Permalink
Rename survey_name field to avoid overloading.
Browse files Browse the repository at this point in the history
  • Loading branch information
alserene committed Nov 2, 2023
1 parent 4a03a45 commit 7eab0a8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions schedview/compute/scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -378,7 +378,7 @@ def get_survey_name(row):
survey_name = make_unique_survey_name(scheduler, [row.list_index, row.survey_index])
return survey_name

summary_df["survey_name"] = summary_df.apply(get_survey_name, axis=1)
summary_df["survey_name_with_id"] = summary_df.apply(get_survey_name, axis=1)

def get_survey_url(row):
if isinstance(row.survey_class, str):
Expand All @@ -399,6 +399,6 @@ 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", "survey_url"]).apply(make_survey_row)
survey_df = summary_df.groupby(["tier", "survey_name_with_id", "survey_url"]).apply(make_survey_row)

return survey_df["reward"].reset_index()

0 comments on commit 7eab0a8

Please sign in to comment.