From bf1c8a3a68c30e8b378ff29a9485447a63525516 Mon Sep 17 00:00:00 2001 From: gcroci2 Date: Mon, 30 Oct 2023 15:43:42 +0100 Subject: [PATCH] fix bug in trainer for testing cases with no target --- deeprank2/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeprank2/trainer.py b/deeprank2/trainer.py index 0a9a92892..d4a3ead8c 100644 --- a/deeprank2/trainer.py +++ b/deeprank2/trainer.py @@ -749,7 +749,7 @@ def _eval( # pylint: disable=too-many-locals count_predictions += pred.shape[0] sum_of_losses += loss_.detach().item() * pred.shape[0] else: - target_vals = ['None'] * pred.shape[0] + target_vals += ['None'] * pred.shape[0] eval_loss = 'None' # Get the outputs for export