diff --git a/stable_baselines3/common/distributions.py b/stable_baselines3/common/distributions.py index a948c4b..532c2e6 100644 --- a/stable_baselines3/common/distributions.py +++ b/stable_baselines3/common/distributions.py @@ -58,7 +58,8 @@ class Distribution(ABC): """ Returns Shannon's entropy of the probability - :return: the entropy, or None if no analytical form is known + :return: The entropy + :raises NotImplementedError: If no analytical form is known """ @abstractmethod