Skip to content

Commit

Permalink
only keep frechet, rmse, and mae to avoid a warning
Browse files Browse the repository at this point in the history
  • Loading branch information
sgbaird committed Jan 7, 2023
1 parent 0f150dc commit d7758f8
Showing 1 changed file with 1 addition and 16 deletions.
17 changes: 1 addition & 16 deletions src/self_driving_lab_demo/utils/search.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,22 +43,7 @@ def evaluation_function(parameters):
dict(R=parameters["R"], G=parameters["G"], B=parameters["B"])
)
# Ax doesn't like the nested dictionary nor a flattened dict with string data
keep_keys = {
"utc_timestamp",
"ch470",
"ch410",
"ch440",
"sd_card_ready",
"ch510",
"ch550",
"ch670",
"onboard_temperature_K",
"ch620",
"ch583",
"frechet",
"rmse",
"mae",
}
keep_keys = {"frechet", "rmse", "mae"}
drop_keys = list(set(results.keys()) - keep_keys)
[results.pop(key, None) for key in drop_keys]
return results
Expand Down

0 comments on commit d7758f8

Please sign in to comment.