Update

[ghstack-poisoned]
pytorch · Jan 9, 2025 · d6bb712 · d6bb712
1 parent eae2413
commit d6bb712
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensordict/_unbatched.py b/tensordict/_unbatched.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 functools import wraps
 from typing import Any, Callable