Update

[ghstack-poisoned]

Author: vmoens

torchrl/envs/transforms/rlhf.py

@@ -2,6 +2,8 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
+
 from copy import copy, deepcopy
 
 import torch

torchrl/envs/transforms/transforms.py

@@ -19,13 +19,9 @@
 from typing import (
     Any,
     Callable,
-    Dict,
-    List,
     Mapping,
-    Optional,
     OrderedDict,
     Sequence,
-    Tuple,
     TypeVar,
     Union,
 )