Update

[ghstack-poisoned]

torchrl/envs/custom/chess.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 Dict, Optional
 
 import torch