Fix typo 2
This commit is contained in:
parent
4a7fbc2470
commit
1c3d3cf6cf
@ -24,9 +24,10 @@ def _patched_make_proba_distribution(
|
||||
cls = StateDependentNoiseDistribution
|
||||
elif use_pca:
|
||||
cls = PCA_Distribution
|
||||
return cls(get_action_dim(action_space), n_envs=n_envs, **dist_kwargs)
|
||||
else:
|
||||
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):
|
||||
return CategoricalDistribution(action_space.n, **dist_kwargs)
|
||||
elif isinstance(action_space, spaces.MultiDiscrete):
|
||||
|
Loading…
Reference in New Issue
Block a user