dodox/metastable-baselines2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix typo 2

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-03-09 13:58:56 +01:00
parent 4a7fbc2470
commit 1c3d3cf6cf
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
metastable_baselines2/common/distributions.py
View File

@ -24,9 +24,10 @@ def _patched_make_proba_distribution(
cls = StateDependentNoiseDistribution cls = StateDependentNoiseDistribution
elif use_pca: elif use_pca:
cls = PCA_Distribution cls = PCA_Distribution
return cls(get_action_dim(action_space), n_envs=n_envs, **dist_kwargs)
else: else:
cls = DiagGaussianDistribution cls = DiagGaussianDistribution
return cls(get_action_dim(action_space), n_envs=n_envs, **dist_kwargs) return cls(get_action_dim(action_space), **dist_kwargs)
elif isinstance(action_space, spaces.Discrete): elif isinstance(action_space, spaces.Discrete):
return CategoricalDistribution(action_space.n, **dist_kwargs) return CategoricalDistribution(action_space.n, **dist_kwargs)
elif isinstance(action_space, spaces.MultiDiscrete): elif isinstance(action_space, spaces.MultiDiscrete):