on_{validation,test}_epoch_end functions should have an outputs parameter
🚀 Feature

  
    
      pytorch-lightning/pytorch_lightning/core/hooks.py
    
    
         Line 255
      in
      3b0e4e0
    
  
  
    

        
          
           def on_validation_epoch_end(self) -> None: 
        
    
  



  
    
      pytorch-lightning/pytorch_lightning/core/hooks.py
    
    
         Line 267
      in
      3b0e4e0
    
  
  
    

        
          
           def on_test_epoch_end(self) -> None: 
        
    
  


Should have an outputs parameter as

  
    
      pytorch-lightning/pytorch_lightning/core/hooks.py
    
    
         Line 243
      in
      3b0e4e0
    
  
  
    

        
          
           def on_train_epoch_end(self, outputs) -> None: