From 6d2c449fafe2ef6e363961dd60e644672a2c862a Mon Sep 17 00:00:00 2001 From: MustafazadaAghasalim Date: Fri, 2 Oct 2026 14:36:10 +0200 Subject: [PATCH] Add a return type to patchcore.score --- src/edd/patchcore.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/edd/patchcore.py b/src/edd/patchcore.py index 4a1f541..605ba26 100644 --- a/src/edd/patchcore.py +++ b/src/edd/patchcore.py @@ -112,7 +112,7 @@ def coreset(x: torch.Tensor, frac: float, dev: torch.device, seed: int = 0) -> t @torch.no_grad() -def score(bank: torch.Tensor, test: torch.Tensor, dev: torch.device, chunk: int = 16): +def score(bank: torch.Tensor, test: torch.Tensor, dev: torch.device, chunk: int = 16) -> torch.Tensor: """Per-patch distance to the nearest normal patch. -> (N, P)""" bank = bank.to(dev) out = []