Skip to content

Commit 41ac9e2

Browse files
Bordalexierule
authored andcommitted
fix TM import MAPE
1 parent a0d29ed commit 41ac9e2

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

pytorch_lightning/metrics/functional/mean_relative_error.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,12 +13,16 @@
1313
# limitations under the License.
1414

1515
import torch
16-
from torchmetrics.functional.regression.mean_relative_error import mean_relative_error as _mean_relative_error
16+
17+
try: # legacy code
18+
from torchmetrics.functional.regression.mean_relative_error import mean_relative_error as _mape
19+
except ImportError:
20+
from torchmetrics.functional.regression import mean_absolute_percentage_error as _mape
1721

1822
from pytorch_lightning.metrics.utils import deprecated_metrics, void
1923

2024

21-
@deprecated_metrics(target=_mean_relative_error)
25+
@deprecated_metrics(target=_mape)
2226
def mean_relative_error(preds: torch.Tensor, target: torch.Tensor) -> torch.Tensor:
2327
"""
2428
.. deprecated::

0 commit comments

Comments
 (0)