Tests: fix deprecated TM mape (#8830)

(cherry picked from commit 3096ab8)

Author: Borda

tests/callbacks/test_quantization.py

@@ -16,10 +16,10 @@
 
 import pytest
 import torch
+from torchmetrics.functional import mean_absolute_percentage_error as mape
 
 from pytorch_lightning import seed_everything, Trainer
 from pytorch_lightning.callbacks import QuantizationAwareTraining
-from pytorch_lightning.metrics.functional.mean_relative_error import mean_relative_error
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from tests.helpers.datamodules import RegressDataModule
 from tests.helpers.runif import RunIf
@@ -41,7 +41,7 @@ def test_quantization(tmpdir, observe: str, fuse: bool, convert: bool):
     trainer = Trainer(**trainer_args)
     trainer.fit(model, datamodule=dm)
     org_size = model.model_size
-    org_score = torch.mean(torch.tensor([mean_relative_error(model(x), y) for x, y in dm.test_dataloader()]))
+    org_score = torch.mean(torch.tensor([mape(model(x), y) for x, y in dm.test_dataloader()]))
 
     fusing_layers = [(f"layer_{i}", f"layer_{i}a") for i in range(3)] if fuse else None
     qcb = QuantizationAwareTraining(observer_type=observe, modules_to_fuse=fusing_layers, quantize_on_fit_end=convert)
@@ -50,7 +50,7 @@ def test_quantization(tmpdir, observe: str, fuse: bool, convert: bool):
 
     quant_calls = qcb._forward_calls
     assert quant_calls == qcb._forward_calls
-    quant_score = torch.mean(torch.tensor([mean_relative_error(qmodel(x), y) for x, y in dm.test_dataloader()]))
+    quant_score = torch.mean(torch.tensor([mape(qmodel(x), y) for x, y in dm.test_dataloader()]))
     # test that the test score is almost the same as with pure training
     assert torch.allclose(org_score, quant_score, atol=0.45)
     model_path = trainer.checkpoint_callback.best_model_path
@@ -69,7 +69,7 @@ def test_quantization(tmpdir, observe: str, fuse: bool, convert: bool):
 
     # todo: make it work also with strict loading
     qmodel2 = RegressionModel.load_from_checkpoint(model_path, strict=False)
-    quant2_score = torch.mean(torch.tensor([mean_relative_error(qmodel2(x), y) for x, y in dm.test_dataloader()]))
+    quant2_score = torch.mean(torch.tensor([mape(qmodel2(x), y) for x, y in dm.test_dataloader()]))
     assert torch.allclose(org_score, quant2_score, atol=0.45)