saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-16 22:20:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Transfer ABC inheritance from BaseModel to BasePolicy (#1062)

Browse source 
Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
This commit is contained in:
Quentin Gallouédec 2022-09-19 22:10:22 +02:00 committed by GitHub
parent d0b129ecc3
commit b7456392ac
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
stable_baselines3/common/policies.py
View file

@ -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.