Fix docstring

Docstring for lightglue forward has leftover doc about `log_assignment`, which is not outputed anymore since refactor.
cvg · Jan 26, 2024 · 3d5e4d0 · 3d5e4d0
1 parent 4959b59
commit 3d5e4d0
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/lightglue/lightglue.py b/lightglue/lightglue.py
@@ -459,7 +459,6 @@ def forward(self, data: dict) -> dict:
                 descriptors: [B x N x D]
                 image: [B x C x H x W] or image_size: [B x 2]
         Output (dict):
-            log_assignment: [B x M+1 x N+1]
             matches0: [B x M]
             matching_scores0: [B x M]
             matches1: [B x N]