target logits bug fix

Author: ViktorDomazetoski

speciesnet/classifier.py

@@ -240,9 +240,9 @@ def batch_predict(
         scores = torch.softmax(logits, dim=-1)
         scores, indices = torch.topk(scores, k=5, dim=-1)
 
-        for filepath, scores_arr, indices_arr in zip(
+        for file_idx, (filepath, scores_arr, indices_arr) in enumerate(zip(
             inference_filepaths, scores.numpy(), indices.numpy()
-        ):
+        )):
 
             predictions[filepath] = {
                 "filepath": filepath,
@@ -257,7 +257,7 @@ def batch_predict(
                     {
                         "target_classes": self.target_labels,
                         "target_logits": [
-                            float(logits[0][idx]) for idx in self.target_idx
+                            float(logits[file_idx][idx]) for idx in self.target_idx
                         ],
                     }
                 )