ImportError: cannot import name 'trim_batch'
🐛 Bug

To Reproduce
Steps to reproduce the behavior:

Go to 'https://colab.research.google.com/drive/1em4d2iTwF1x4yZ10J5g5LtLcXBwgiyTA?usp=sharing'
Run
Scroll down to 'LN 26'
See error

Environment
Please copy and paste the output from our

CUDA:

GPU:

Tesla T4


available:         True
version:           10.1


Packages:

numpy:             1.18.5
pyTorch_debug:     False
pyTorch_version:   1.6.0+cu101
pytorch-lightning: 0.9.0
tensorboard:       2.2.0
tqdm:              4.41.1


System:

OS:                Linux
architecture:

64bit



processor:         x86_64
python:            3.6.9
version:           #1 SMP Thu Jul 23 08:00:38 PDT 2020