change method signature for better pycharm auto complete
🚀 Feature
imporve pycharm auto complete experience
Motivation
The auto complete result when typing def training_step
def training_step(self, *args, **kwargs):

I want it to be
def training_step(self, batch, batch_idx):

Many Thanks