[bug] needs to be super() for eval

uhlmanngroup · Aug 15, 2024 · 35820cf · 35820cf
1 parent fec2ee3
commit 35820cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bioimage_embed/shapes/lightning.py b/bioimage_embed/shapes/lightning.py
@@ -32,8 +32,8 @@ def batch_to_tensor(self, batch):
         return output
 
     def eval_step(self, batch, batch_idx):
-        # model_output = super().eval_step(batch, batch_idx)
-        model_output = self.predict_step(batch, batch_idx)
+        # Needs to be super because eval_step is overwritten in Supervised
+        model_output = super().eval_step(batch, batch_idx)
         loss_ops = lf.DistanceMatrixLoss(model_output.recon_x, norm=False)
 
         shape_loss = torch.sum(