Remove MaxAE from TorchANI (#150)

torchani/ignite.py

@@ -6,6 +6,7 @@ from . import utils
 from torch.nn.modules.loss import _Loss
 from ignite.metrics.metric import Metric
 from ignite.metrics import RootMeanSquaredError
+from ignite.contrib.metrics.regression import MaximumAbsoluteError
 
 
 class Container(torch.nn.ModuleDict):
@@ -111,29 +112,9 @@ def RMSEMetric(key):
     return DictMetric(key, RootMeanSquaredError())
 
 
-class MaxAbsoluteError(Metric):
-    """
-    Calculates the max absolute error.
-
-    - `update` must receive output of the form `(y_pred, y)`.
-    """
-    def reset(self):
-        self._max_of_absolute_errors = 0.0
-
-    def update(self, output):
-        y_pred, y = output
-        absolute_errors = torch.abs(y_pred - y.view_as(y_pred))
-        batch_max = absolute_errors.max().item()
-        if batch_max > self._max_of_absolute_errors:
-            self._max_of_absolute_errors = batch_max
-
-    def compute(self):
-        return self._max_of_absolute_errors
-
-
 def MAEMetric(key):
     """Create max absolute error metric on key."""
-    return DictMetric(key, MaxAbsoluteError())
+    return DictMetric(key, MaximumAbsoluteError())
 
 
 __all__ = ['Container', 'MSELoss', 'TransformedLoss', 'RMSEMetric',