diff --git a/tensorflow_recommenders/tasks/ranking.py b/tensorflow_recommenders/tasks/ranking.py
index 5822753..6302bf0 100644
--- a/tensorflow_recommenders/tasks/ranking.py
+++ b/tensorflow_recommenders/tasks/ranking.py
@@ -70,7 +70,6 @@ def call(self,
            labels: tf.Tensor,
            predictions: tf.Tensor,
            sample_weight: Optional[tf.Tensor] = None,
-           training: bool = False,
            compute_metrics: bool = True) -> tf.Tensor:
     """Computes the task loss and metrics.
 
@@ -78,7 +77,6 @@ def call(self,
       labels: Tensor of labels.
       predictions: Tensor of predictions.
       sample_weight: Tensor of sample weights.
-      training: Indicator whether training or test loss is being computed.
       compute_metrics: Whether to compute metrics. Set this to False
         during training for faster training.