docs: Update implement.rst (#2345)

Lightning-AI · Feb 7, 2024 · 2c7ccfe · 2c7ccfe
1 parent dc115fe
commit 2c7ccfe
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)