fix(rllib): allow explore=False with tuple action distributions (#10443)

This commit is contained in:
Thomas Lecat authored and GitHub committed 2020-09-10 15:03:02 -07:00
1 parent 7aa55ca82c
commit 504da45e69
1 file changed
+1 -1
@@ -72,5 +72,5 @@ class StochasticSampling(Exploration):
logp = action_dist.sampled_action_logp()
else:
action = action_dist.deterministic_sample()
logp = torch.zeros((action.size()[0], ), dtype=torch.float32)
logp = torch.zeros_like(action_dist.sampled_action_logp())
return action, logp