mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-07-30 20:18:15 +00:00
parent
c69f7cd5e6
commit
6f822b9ed7
3 changed files with 4 additions and 3 deletions
|
|
@ -1,4 +1,4 @@
|
|||
image: stablebaselines/stable-baselines3-cpu:0.11.1
|
||||
image: stablebaselines/stable-baselines3-cpu:1.1.0a5
|
||||
|
||||
type-check:
|
||||
script:
|
||||
|
|
|
|||
|
|
@ -33,6 +33,7 @@ Others:
|
|||
- Added ``flake8-bugbear`` to tests dependencies to find likely bugs
|
||||
- Added Code of Conduct
|
||||
- Added tests for GAE and lambda return computation
|
||||
- Updated docker image with newest black version
|
||||
|
||||
Documentation:
|
||||
^^^^^^^^^^^^^^
|
||||
|
|
|
|||
|
|
@ -104,7 +104,7 @@ class BaseModel(nn.Module, ABC):
|
|||
return net_kwargs
|
||||
|
||||
def make_features_extractor(self) -> BaseFeaturesExtractor:
|
||||
""" Helper method to create a features extractor."""
|
||||
"""Helper method to create a features extractor."""
|
||||
return self.features_extractor_class(self.observation_space, **self.features_extractor_kwargs)
|
||||
|
||||
def extract_features(self, obs: th.Tensor) -> th.Tensor:
|
||||
|
|
@ -203,7 +203,7 @@ class BasePolicy(BaseModel):
|
|||
|
||||
@staticmethod
|
||||
def _dummy_schedule(progress_remaining: float) -> float:
|
||||
""" (float) Useful for pickling policy."""
|
||||
"""(float) Useful for pickling policy."""
|
||||
del progress_remaining
|
||||
return 0.0
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue