fix bug in trainer for testing cases with no target

DeepRank · Oct 30, 2023 · bf1c8a3 · bf1c8a3
1 parent 10a5795
commit bf1c8a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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