[BugFix] Fix typing in llm env

ghstack-source-id: b5608f91756b5a81141941903158417a111e0710 Pull Request resolved: pytorch#2647
kurtamohler · Dec 13, 2024 · e3c3047 · e3c3047
1 parent cb8e241
commit e3c3047
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchrl/envs/custom/llm.py b/torchrl/envs/custom/llm.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 typing import Callable, List, Union
 
 import torch