Remove deprecated argument (#14)

PRBonn · Apr 20, 2024 · 82d7bf6 · 82d7bf6
1 parent 805526c
commit 82d7bf6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pcf/models/base.py b/pcf/models/base.py
@@ -207,7 +207,7 @@ def test_step(self, batch, batch_idx):
 
         return loss
 
-    def on_test_epoch_end(self, outputs):
+    def on_test_epoch_end(self):
         # Remove first row since it was initialized with zero
         self.chamfer_distances_tensor = self.chamfer_distances_tensor[:, 1:]
         n_steps, _ = self.chamfer_distances_tensor.shape