Fix import

pytorch · May 29, 2024 · 05e2367 · 05e2367
1 parent f5ffcd4
commit 05e2367
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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())