🐛 Fix Expected a parent bug, see [the solution](Lightning-AI/pytorch-…

…lightning#17485 (comment))

engine/callbacks.py

@@ -1,4 +1,4 @@
-from pytorch_lightning.callbacks import Callback
+from lightning.pytorch.callbacks import Callback
 import numpy as np
 import pandas as pd
 from datetime import date
@@ -19,11 +19,4 @@ def on_test_end(self, *args, **kwargs):
         test_info = test_info.reset_index().rename(columns={"index": "ID"})
         file_name = f"{self.model_name}_{date.today()}.csv"
         test_info.to_csv(file_name, index=False, lineterminator='\n')
-        print("Output csv file successfully saved!!")
-
-
-    def load_state_dict(self, state_dict):
-        self.predictions = state_dict['predictions']
-
-    def state_dict(self):
-        return {'predictions': self.predictions}
+        print("Output csv file successfully saved!!")