Training notebook fails from pytorch-lightning "unexpected keyword argument"

[gaito-20]

Describe the bug
The given google colab notebook for the trainer fails

To Reproduce
Steps to reproduce the behavior:

1. Go to the Trainer Google Colab
2. Execute the cells
3. 2nd cell fails at model.train(train_df=train_df[:100], test_df=test_df[:50], batch_size=2, max_epochs=3,use_gpu=True)
4. See error

Expected behavior
An initial execution to succeed.

Screenshots

[AstroNoodles]

Has this been updated yet? Just ran the Trainer.ipynb notebook and got the same error. Seems to be hidden deep down in how Lightning works