[fix] argument in evaluator

Author: jianwensong

compressai_vision/evaluators/evaluators.py

@@ -1453,7 +1453,7 @@ def write_results(self, path: str = None):
         with open(f"{path}/{self.output_file_name}.json", "w", encoding="utf-8") as f:
             json.dump(self._evaluations, f, ensure_ascii=False, indent=4)
 
-    def digest(self, gt, pred):
+    def digest(self, gt, pred, mse_results=None):
         ref = gt[0]["image"].unsqueeze(0).cpu()
         tst = pred.unsqueeze(0).cpu()