AttributeError: 'DistributedDataParallel' object has no attribute 'test_step'

[esraashabrani]

🐛 Bug

I was trying the K-Fold cross validation example and got this error.

To Reproduce

Run this code as a python script https://github.com/Lightning-AI/lightning/blob/master/examples/pl_loops/kfold.py.

Expected behavior

Environment

• Lightning Component (e.g. Trainer, LightningModule, LightningApp, LightningWork, LightningFlow):
• PyTorch Lightning Version (1.6.5)
• torch Version (1.12.0+cu113)
• torchvision Version (0.13.0+cu113)
• Python version (3.8)

Additional context

[esraashabrani]

Any updates please? @awaelchli

[elifdaldal]

I am having the same error. I'm running Python 3.9.13.