[BugFix] Fix typing in llm env

Vincent Moens · Vincent Moens · commit e3c304733757 · 2024-12-13T13:55:21.000-08:00
ghstack-source-id: b5608f9 Pull Request resolved: #2647
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

Original file line number	Diff line number	Diff line change
`@@ -2,6 +2,8 @@`
`2`	`2`	`#`
`3`	`3`	`# This source code is licensed under the MIT license found in the`
`4`	`4`	`# LICENSE file in the root directory of this source tree.`
	`5`	`+from __future__ import annotations`
	`6`	`+`
`5`	`7`	`from typing import Callable, List, Union`
`6`	`8`
`7`	`9`	`import torch`