From 93f849a5c51b963894d2fd3d338ed755fdb3770b Mon Sep 17 00:00:00 2001 From: aws-kingrj <78175353+aws-kingrj@users.noreply.github.com> Date: Wed, 8 Nov 2023 09:35:41 -0800 Subject: [PATCH] fixing neuron import (#5775) --- torch_xla/_internal/neuron.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torch_xla/_internal/neuron.py b/torch_xla/_internal/neuron.py index 2f286681136..7b145918675 100644 --- a/torch_xla/_internal/neuron.py +++ b/torch_xla/_internal/neuron.py @@ -1,4 +1,5 @@ import os +import logging def num_local_processes() -> int: