import contextlib

pytorch · May 16, 2024 · e69279d · e69279d
1 parent 1a496c0
commit e69279d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch_xla/torch_xla.py b/torch_xla/torch_xla.py
@@ -1,4 +1,6 @@
+import contextlib
 from typing import List
+
 import torch
 import torch_xla
 import torch_xla.core.xla_model as xm