diff --git a/mo_gymnasium/envs/minecart/minecart.py b/mo_gymnasium/envs/minecart/minecart.py index bf5052ff..568a14f9 100644 --- a/mo_gymnasium/envs/minecart/minecart.py +++ b/mo_gymnasium/envs/minecart/minecart.py @@ -712,7 +712,7 @@ def distribution_means(self): means = np.zeros(len(self.distributions)) for i, dist in enumerate(self.distributions): - mean, std = dist.mean(), dist.std() + mean, std = dist.args means[i] = truncated_mean(mean, std, 0, float("inf")) if np.isnan(means[i]): means[i] = 0