Fix WEIGHTED_NUM_SAMPES typo in torchrec/metrics (meta-pytorch#3489)

Summary:
Pull Request resolved: meta-pytorch#3489

Fix typo: WEIGHTED_NUM_SAMPES -> WEIGHTED_NUM_SAMPLES

Reviewed By: iamzainhuda

Differential Revision: D85461614

fbshipit-source-id: 77c6e4a70fdc0bab5fb8f796c2193c262cc99818

Author: Jinpu Zhou

torchrec/metrics/mae.py

@@ -20,7 +20,7 @@
 
 
 ERROR_SUM = "error_sum"
-WEIGHTED_NUM_SAMPES = "weighted_num_samples"
+WEIGHTED_NUM_SAMPLES = "weighted_num_samples"
 
 
 def compute_mae(
@@ -106,7 +106,7 @@ def _compute(self) -> List[MetricComputationReport]:
                 metric_prefix=MetricPrefix.WINDOW,
                 value=compute_mae(
                     self.get_window_state(ERROR_SUM),
-                    self.get_window_state(WEIGHTED_NUM_SAMPES),
+                    self.get_window_state(WEIGHTED_NUM_SAMPLES),
                 ),
             ),
         ]

torchrec/metrics/mse.py

@@ -21,7 +21,7 @@
 
 
 ERROR_SUM = "error_sum"
-WEIGHTED_NUM_SAMPES = "weighted_num_samples"
+WEIGHTED_NUM_SAMPLES = "weighted_num_samples"
 LABEL_SUM = "label_sum"
 LABEL_SQUARED_SUM = "label_squared_sum"
 
@@ -162,15 +162,15 @@ def _compute(self) -> List[MetricComputationReport]:
                 metric_prefix=MetricPrefix.WINDOW,
                 value=compute_mse(
                     self.get_window_state(ERROR_SUM),
-                    self.get_window_state(WEIGHTED_NUM_SAMPES),
+                    self.get_window_state(WEIGHTED_NUM_SAMPLES),
                 ),
             ),
             MetricComputationReport(
                 name=MetricName.RMSE,
                 metric_prefix=MetricPrefix.WINDOW,
                 value=compute_rmse(
                     self.get_window_state(ERROR_SUM),
-                    self.get_window_state(WEIGHTED_NUM_SAMPES),
+                    self.get_window_state(WEIGHTED_NUM_SAMPLES),
                 ),
             ),
         ]
@@ -192,7 +192,7 @@ def _compute(self) -> List[MetricComputationReport]:
                     metric_prefix=MetricPrefix.WINDOW,
                     value=compute_r_squared(
                         self.get_window_state(ERROR_SUM),
-                        self.get_window_state(WEIGHTED_NUM_SAMPES),
+                        self.get_window_state(WEIGHTED_NUM_SAMPLES),
                         self.get_window_state(LABEL_SUM),
                         self.get_window_state(LABEL_SQUARED_SUM),
                     ),

torchrec/metrics/nmse.py

@@ -19,7 +19,7 @@
     ERROR_SUM,
     get_mse_states,
     MSEMetricComputation,
-    WEIGHTED_NUM_SAMPES,
+    WEIGHTED_NUM_SAMPLES,
 )
 from torchrec.metrics.rec_metric import (
     MetricComputationReport,
@@ -117,21 +117,21 @@ def _compute(self) -> List[MetricComputationReport]:
 
         window_mse = compute_mse(
             self.get_window_state(ERROR_SUM),
-            self.get_window_state(WEIGHTED_NUM_SAMPES),
+            self.get_window_state(WEIGHTED_NUM_SAMPLES),
         )
         window_const_pred_mse = compute_mse(
             self.get_window_state(CONST_PRED_ERROR_SUM),
-            self.get_window_state(WEIGHTED_NUM_SAMPES),
+            self.get_window_state(WEIGHTED_NUM_SAMPLES),
         )
         window_nmse = compute_norm(window_mse, window_const_pred_mse)
 
         window_rmse = compute_rmse(
             self.get_window_state(ERROR_SUM),
-            self.get_window_state(WEIGHTED_NUM_SAMPES),
+            self.get_window_state(WEIGHTED_NUM_SAMPLES),
         )
         window_const_pred_rmse = compute_rmse(
             self.get_window_state(CONST_PRED_ERROR_SUM),
-            self.get_window_state(WEIGHTED_NUM_SAMPES),
+            self.get_window_state(WEIGHTED_NUM_SAMPLES),
         )
         window_nrmse = compute_norm(window_rmse, window_const_pred_rmse)