Fix (#3180)

matteobettini · web-flow · commit 45bd3c05aebe · 2025-10-02T12:34:08.000+01:00
diff --git a/torchrl/modules/distributions/discrete.py b/torchrl/modules/distributions/discrete.py
@@ -172,7 +172,7 @@ def rsample(self, sample_shape: torch.Size | Sequence = None) -> torch.Tensor:
             )
 
 
-class (D.Categorical):
+class MaskedCategorical(D.Categorical):
     """MaskedCategorical distribution.
 
     Reference:

Original file line number	Diff line number	Diff line change
`@@ -172,7 +172,7 @@ def rsample(self, sample_shape: torch.Size \| Sequence = None) -> torch.Tensor:`
`172`	`172`	`)`
`173`	`173`
`174`	`174`
`175`		`-class (D.Categorical):`
	`175`	`+class MaskedCategorical(D.Categorical):`
`176`	`176`	`"""MaskedCategorical distribution.`
`177`	`177`
`178`	`178`	`Reference:`