diff --git a/examples/redq/utils.py b/examples/redq/utils.py index bcf907314b9..9c8015724a9 100644 --- a/examples/redq/utils.py +++ b/examples/redq/utils.py @@ -568,7 +568,7 @@ def transformed_env_constructor( Args: cfg (DictConfig): a DictConfig containing the arguments of the script. video_tag (str, optional): video tag to be passed to the Logger object - logger (Logger, optional): torchrl_logger associated with the script + logger (Logger, optional): logger associated with the script stats (dict, optional): a dictionary containing the :obj:`loc` and :obj:`scale` for the `ObservationNorm` transform norm_obs_only (bool, optional): If `True` and `VecNorm` is used, the reward won't be normalized online. Default is `False`. diff --git a/torchrl/trainers/helpers/envs.py b/torchrl/trainers/helpers/envs.py index 36a724127a6..8dc8d478ddf 100644 --- a/torchrl/trainers/helpers/envs.py +++ b/torchrl/trainers/helpers/envs.py @@ -227,7 +227,7 @@ def transformed_env_constructor( Args: cfg (DictConfig): a DictConfig containing the arguments of the script. video_tag (str, optional): video tag to be passed to the Logger object - logger (Logger, optional): torchrl_logger associated with the script + logger (Logger, optional): logger associated with the script stats (dict, optional): a dictionary containing the :obj:`loc` and :obj:`scale` for the `ObservationNorm` transform norm_obs_only (bool, optional): If `True` and `VecNorm` is used, the reward won't be normalized online. Default is `False`.