Use span.get_bbox() instead of get_bbox()

HazyResearch · Jun 13, 2020 · 81bb196 · 81bb196
1 parent 31c4b37
commit 81bb196
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/fonduer/utils/visualizer.py b/src/fonduer/utils/visualizer.py
@@ -78,9 +78,7 @@ def display_candidates(
                 pdf_file += ".PDF"
             else:
                 logger.error("display_candidates failed: pdf file missing.")
-        boxes = [
-            get_box(mention.context) for c in candidates for mention in c.get_mentions()
-        ]
+        boxes = [m.context.get_bbox() for c in candidates for m in c.get_mentions()]
         imgs = self.display_boxes(pdf_file, boxes, alternate_colors=True)
         return display(*imgs)