diff --git a/tensordict/nn/distributions/discrete.py b/tensordict/nn/distributions/discrete.py index 81db384dc..78218d4ee 100644 --- a/tensordict/nn/distributions/discrete.py +++ b/tensordict/nn/distributions/discrete.py @@ -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,