Fix attribute error and multigpu inference (#177, #178)

RicardoRei · RicardoRei · commit 2bcf66604b30 · 2024-01-08T14:08:04.000Z
diff --git a/comet/models/predict_writer.py b/comet/models/predict_writer.py
@@ -86,7 +86,7 @@ def flatten(list):
 
         def flatten_predictions(predictions):
             flatten_pred = Prediction(
-                scores=torch.cat([pred.scores for pred in predictions], dim=0)
+                scores=torch.cat([pred["scores"] for pred in predictions], dim=0)
             )
             if "metadata" in predictions[0]:
                 flatten_pred["metadata"] = flatten_metadata(

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,7 @@ def flatten(list):`
`86`	`86`
`87`	`87`	`def flatten_predictions(predictions):`
`88`	`88`	`flatten_pred = Prediction(`
`89`		`- scores=torch.cat([pred.scores for pred in predictions], dim=0)`
	`89`	`+ scores=torch.cat([pred["scores"] for pred in predictions], dim=0)`
`90`	`90`	`)`
`91`	`91`	`if "metadata" in predictions[0]:`
`92`	`92`	`flatten_pred["metadata"] = flatten_metadata(`