If I can make a suggestion, perhaps you could rename the function mushroom_rl.utils.dataset.episodes_length() to mushroom_rl.utils.dataset.compute_episodes_length(). This would better match your function naming convention (e.g. compute_J(), compute_metrics()) and would also allow me to name the returned variable episode_lengths.
If I can make a suggestion, perhaps you could rename the function
mushroom_rl.utils.dataset.episodes_length()tomushroom_rl.utils.dataset.compute_episodes_length(). This would better match your function naming convention (e.g.compute_J(),compute_metrics()) and would also allow me to name the returned variableepisode_lengths.Thanks in advance!