Compatibility with python < 3.10

pytorch · Mar 7, 2024 · 5a854c5 · 5a854c5
1 parent 6bd9dc6
commit 5a854c5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchsnapshot/io_preparers/sharded_tensor.py b/torchsnapshot/io_preparers/sharded_tensor.py
@@ -4,6 +4,7 @@
 #
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
 
 import asyncio
 import copy