Custom metrics only mean value

Hey, I am logging custom metrics from my ray tune run to tensorboard by overriding the on_episode_end function from DefaultCallbacks . For each value I have currently value_mean, value_max and value_min, but I only want to save the value_mean field. Is there a way to configure this? (edited)

No, unfortunately, this is still pretty hard-coded to store mean/max/min values only.