amend

pytorch · Jan 31, 2024 · f32ce83 · f32ce83
1 parent c8d6441
commit f32ce83
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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`.