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: