ImportError: cannot import name 'inf' from 'torch'

[pankaj-kvhld]

I am trying it on Kaggle.

I ran pip install fastai-xla-extensions==0.0.6 to install library successfully.

But on running
import torch_xla

I get

ImportError                               Traceback (most recent call last)
Cell In[4], line 1
----> 1 import torch_xla

File /usr/local/lib/python3.8/site-packages/torch_xla/__init__.py:179
    177 import atexit
    178 import torch
--> 179 from ._patched_functions import _apply_patches
    180 from .version import __version__
    182 logger.info(
    183     'Letting libtpu.so load fail during _XLAC import. libtpu.so will be loaded '
    184     'from `libtpu` Python package when the ComputationClient is created.')

File /usr/local/lib/python3.8/site-packages/torch_xla/_patched_functions.py:4
      2 import torch
      3 import torch.nn as nn
----> 4 from torch import inf
      5 from typing import Iterable, Union, Optional
      7 _tensor_or_tensors = Union[torch.Tensor, Iterable[torch.Tensor]]

ImportError: cannot import name 'inf' from 'torch' (/usr/local/lib/python3.8/site-packages/torch/__init__.py)

[mphetens]

Markdown_1.0.1.zip