From 5940d6bc9c1c5598b9fa570596b433513571be28 Mon Sep 17 00:00:00 2001
From: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
Date: Tue, 6 Dec 2022 14:02:12 +0800
Subject: [PATCH] [Fix] textspotting ut (#1599)

---
 tests/test_visualization/test_textspotting_visualizer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_visualization/test_textspotting_visualizer.py b/tests/test_visualization/test_textspotting_visualizer.py
index 91086475..5c064339 100644
--- a/tests/test_visualization/test_textspotting_visualizer.py
+++ b/tests/test_visualization/test_textspotting_visualizer.py
@@ -13,7 +13,7 @@ from mmocr.utils import bbox2poly
 from mmocr.visualization import TextSpottingLocalVisualizer
 
 
-class TestTextKIELocalVisualizer(unittest.TestCase):
+class TestTextSpottingLocalVisualizer(unittest.TestCase):
 
     def setUp(self):
         h, w = 12, 10
@@ -95,7 +95,7 @@ class TestTextKIELocalVisualizer(unittest.TestCase):
                 draw_pred=False,
                 out_file=out_file)
             self._assert_image_and_shape(out_file, (h, w * 2, c))
-            bboxes = self.data_sample.pred_instances.pop('bboxes')
+            bboxes = self.data_sample.pred_instances.get('bboxes')
             bboxes = bboxes.tolist()
             polys = [bbox2poly(bbox) for bbox in bboxes]
             self.data_sample.pred_instances.polygons = polys