pt_lib/utils/torchrec_metrics.py changes to combine story rec metrics on uniarch (#875)

Summary:
Pull Request resolved: #875

1. Attach torchrec.metrics.mae to RecMetricEnum.MAE

Differential Revision: D41782396

fbshipit-source-id: f0677cd206c3a8d5c3d0789c10a36e5745ead71c

Author: Shakti Kumar

torchrec/metrics/metric_module.py

@@ -18,6 +18,7 @@
 from torchrec.metrics.auc import AUCMetric
 from torchrec.metrics.calibration import CalibrationMetric
 from torchrec.metrics.ctr import CTRMetric
+from torchrec.metrics.mae import MAEMetric
 from torchrec.metrics.metrics_config import (
     MetricsConfig,
     RecMetricEnum,
@@ -46,6 +47,7 @@
     RecMetricEnum.CALIBRATION: CalibrationMetric,
     RecMetricEnum.AUC: AUCMetric,
     RecMetricEnum.MSE: MSEMetric,
+    RecMetricEnum.MAE: MAEMetric,
     RecMetricEnum.MULTICLASS_RECALL: MulticlassRecallMetric,
 }
 

torchrec/metrics/metrics_config.py

@@ -22,6 +22,7 @@ class RecMetricEnum(RecMetricEnumBase):
     AUC = "auc"
     CALIBRATION = "calibration"
     MSE = "mse"
+    MAE = "mae"
     MULTICLASS_RECALL = "multiclass_recall"