From b7456392acb921f68a6b9f894350c099726e5a55 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Quentin=20Gallou=C3=A9dec?= <45557362+qgallouedec@users.noreply.github.com> Date: Mon, 19 Sep 2022 22:10:22 +0200 Subject: [PATCH] Transfer `ABC` inheritance from `BaseModel` to `BasePolicy` (#1062) Co-authored-by: Antonin RAFFIN --- stable_baselines3/common/policies.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stable_baselines3/common/policies.py b/stable_baselines3/common/policies.py index cd372e1..16afe06 100644 --- a/stable_baselines3/common/policies.py +++ b/stable_baselines3/common/policies.py @@ -34,7 +34,7 @@ from stable_baselines3.common.type_aliases import Schedule from stable_baselines3.common.utils import get_device, is_vectorized_observation, obs_as_tensor -class BaseModel(nn.Module, ABC): +class BaseModel(nn.Module): """ The base model object: makes predictions in response to observations. @@ -251,7 +251,7 @@ class BaseModel(nn.Module, ABC): return observation, vectorized_env -class BasePolicy(BaseModel): +class BasePolicy(BaseModel, ABC): """The base policy object. Parameters are mostly the same as `BaseModel`; additions are documented below.