From d22266d05d7ae10f53e3b904d847d44743beba40 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Valter=20Sch=C3=BCtz?= Date: Tue, 3 Dec 2024 15:12:49 +0100 Subject: [PATCH] [Doc] Update docstring for TruncatedNormal with correct parameter names (#2625) Co-authored-by: Valter Schutz --- torchrl/modules/distributions/continuous.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchrl/modules/distributions/continuous.py b/torchrl/modules/distributions/continuous.py index eb9093dbcfe..e34f1be8ff9 100644 --- a/torchrl/modules/distributions/continuous.py +++ b/torchrl/modules/distributions/continuous.py @@ -205,8 +205,8 @@ class TruncatedNormal(D.Independent): Default is 5.0 - min (torch.Tensor or number, optional): minimum value of the distribution. Default = -1.0; - max (torch.Tensor or number, optional): maximum value of the distribution. Default = 1.0; + low (torch.Tensor or number, optional): minimum value of the distribution. Default = -1.0; + high (torch.Tensor or number, optional): maximum value of the distribution. Default = 1.0; tanh_loc (bool, optional): if ``True``, the above formula is used for the location scaling, otherwise the raw value is kept. Default is ``False``;