def validation_step(self, batch: Batch, *args, **kwargs) -> Batch:
        """Validation steps that returns random image-level scores."""
        del args, kwargs
        batch.pred_score = torch.rand(batch.batch_size, device=self.device)
        return batch
Validation steps that returns random image-level scores.
tests/integration/test_task_types.py