From e3c304733757616bf8bad166e070a53ec1385898 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Fri, 13 Dec 2024 13:55:08 -0800
Subject: [PATCH] [BugFix] Fix typing in llm env

ghstack-source-id: b5608f91756b5a81141941903158417a111e0710
Pull Request resolved: https://github.com/pytorch/rl/pull/2647
---
 torchrl/envs/custom/llm.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/torchrl/envs/custom/llm.py b/torchrl/envs/custom/llm.py
index 2f456482147..4b8b1a5f21b 100644
--- 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