Skip to content

Commit d22266d

Browse files
valterschutzValter Schutz
andauthored
[Doc] Update docstring for TruncatedNormal with correct parameter names (#2625)
Co-authored-by: Valter Schutz <valterschutz@proton.me>
1 parent 8257799 commit d22266d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchrl/modules/distributions/continuous.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -205,8 +205,8 @@ class TruncatedNormal(D.Independent):
205205
206206
Default is 5.0
207207
208-
min (torch.Tensor or number, optional): minimum value of the distribution. Default = -1.0;
209-
max (torch.Tensor or number, optional): maximum value of the distribution. Default = 1.0;
208+
low (torch.Tensor or number, optional): minimum value of the distribution. Default = -1.0;
209+
high (torch.Tensor or number, optional): maximum value of the distribution. Default = 1.0;
210210
tanh_loc (bool, optional): if ``True``, the above formula is used for
211211
the location scaling, otherwise the raw value is kept.
212212
Default is ``False``;

0 commit comments

Comments
 (0)