Skip to content

Commit

Permalink
[BugFix] Fix typo in OneHotCategorical (#1130)
Browse files Browse the repository at this point in the history
  • Loading branch information
joachxm authored Dec 9, 2024
1 parent 5349f2a commit d7506c4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensordict/nn/distributions/discrete.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ def mode(self) -> torch.Tensor:
else:
return (self.probs == self.probs.max(-1, True)[0]).to(torch.long)

determnistic_sample = mode
deterministic_sample = mode

def sample(
self,
Expand Down

0 comments on commit d7506c4

Please sign in to comment.