From e69279dea2d6df41c095d4ce44ad1b5864b4beca Mon Sep 17 00:00:00 2001
From: Will Cromar <wcromar@google.com>
Date: Thu, 16 May 2024 16:23:14 +0000
Subject: [PATCH] import contextlib

---
 torch_xla/torch_xla.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/torch_xla/torch_xla.py b/torch_xla/torch_xla.py
index 5f79fc27f6c..ef8e30667da 100644
--- 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