From 31dd451436c2ad6557fc545772b4a3b14d4f3942 Mon Sep 17 00:00:00 2001
From: Amithrajith Mamidala <amithrm@amazon.com>
Date: Mon, 25 Mar 2024 09:36:50 -0700
Subject: [PATCH] Fix import

---
 torch_xla/core/xla_model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py
index 8f28ab8a4997..cdc4d84195b6 100644
--- a/torch_xla/core/xla_model.py
+++ b/torch_xla/core/xla_model.py
@@ -18,6 +18,7 @@
 import torch_xla.debug.metrics_saver as ms
 import torch_xla.utils.utils as xu
 import torch_xla.utils.closures as xc
+import os
 
 _DEVICES = xu.LazyProperty(lambda: torch_xla._XLAC._xla_get_devices())