docs: Update implement.rst (#2345)

(cherry picked from commit 2c7ccfe)
Lightning-AI · Feb 12, 2024 · b945fd3 · b945fd3
1 parent 4b92498
commit b945fd3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/pages/implement.rst b/docs/source/pages/implement.rst
@@ -98,8 +98,8 @@ because we need to calculate the rank of the predictions and targets.
 
         def compute(self):
             # parse inputs
-            preds = dim_zero_cat(preds)
-            target = dim_zero_cat(target)
+            preds = dim_zero_cat(self.preds)
+            target = dim_zero_cat(self.target)
             # some intermediate computation...
             r_preds, r_target = _rank_data(preds), _rank_dat(target)
             preds_diff = r_preds - r_preds.mean(0)