Can't import pytorch lightening module in colab

[Dheeraj-Arremsetty]

🐛 Bug

Hello Pytorch Team,
I have been working on pytorch lightning for several days, I have never seen this error on Google Colab.
Not sure if there is any change that happened on colab side or lightning.

When I install pytorch lightning and try to import pytorch lightening module, it can't import and throws out the error. The same was working till yesterday. Not sure If I am doing anything wrong on my side, here is the screenshot:

Same for bolts as well.

To Reproduce

Steps to reproduce the behavior:

!pip install pytorch-lightning --quiet
import pytorch_lightning as pl

!pip install pytorch-lightning-bolts --quiet
import pl_bolts as plb

Environment

• PyTorch Version (e.g., 1.0):latest
• OS (e.g., Linux):Colab
• How you installed PyTorch (conda, pip, source): pip
• Build command you used (if compiling from source):
• Python version:
• CUDA/cuDNN version:
• GPU models and configuration:
• Any other relevant information:

Additional context

[kunal-savvy]

I am also getting same error. This is blocking our work. Any help?

[kunal-savvy]

ImportError                               Traceback (most recent call last)

<ipython-input-5-efd8697dde72> in <module>()
----> 1 import pytorch_lightning as pl

9 frames

/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/apply_func.py in <module>()
     26 
     27 if _TORCHTEXT_AVAILABLE:
---> 28     from torchtext.data import Batch
     29 else:
     30     Batch = type(None)

ImportError: cannot import name 'Batch' from 'torchtext.data' (/usr/local/lib/python3.7/dist-packages/torchtext/data/__init__.py)


---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.`
```

[ajsanjoaquin]

Hello,

This is the same problem here. See the last post on the issue for a workaround.

[akihironitta]

@Dheeraj-Arremsetty @kunal-savvy Hi, sorry for the delay. As @ajsanjoaquin mentioned, the problem should be resolved now. Mind having a look at Lightning-AI/pytorch-lightning#6415 or stackoverflow?