Fix smol Bug

This commit is contained in:
Dominik Moritz Roth 2023-05-21 17:15:46 +02:00
parent 834b5c9a65
commit c473990fbd

View File

@ -299,7 +299,7 @@ class ActorCriticPolicy(BasePolicy):
values = self.value_net(latent_vf)
distribution = self._get_action_dist_from_latent(latent_pi)
if self.use_pca:
assert trajectory, 'Past trajetcory has to be provided when using PCA.'
assert trajectory != None, 'Past trajectory has to be provided when using PCA.'
actions = distribution.get_actions(deterministic=deterministic, trajectory=trajectory)
else:
actions = distribution.get_actions(deterministic=deterministic)