From d7506c4964073917b3051de2eff04953555f92ba Mon Sep 17 00:00:00 2001 From: joachxm <155843363+joachxm@users.noreply.github.com> Date: Mon, 9 Dec 2024 23:52:05 +0100 Subject: [PATCH] [BugFix] Fix typo in `OneHotCategorical` (#1130) --- tensordict/nn/distributions/discrete.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,