diff --git a/tensordict/base.py b/tensordict/base.py index bae83ba67..f361c93c8 100644 --- a/tensordict/base.py +++ b/tensordict/base.py @@ -22,8 +22,9 @@ OrderedDict, overload, Sequence, + Tuple, TypeVar, - Union, Tuple, + Union, ) import numpy as np