Forráskód Böngészése

fix save_all dont work in cli

gaotingquan 1 éve
szülő
commit
fc2c8bbf3d
1 módosított fájl, 6 hozzáadás és 1 törlés
  1. 6 1
      paddlex/inference/results/table_rec.py

+ 6 - 1
paddlex/inference/results/table_rec.py

@@ -71,9 +71,14 @@ class StructureTableResult(TableRecResult, HtmlMixin, XlsxMixin):
         return self["html"]
 
 
-class TableResult(BaseResult):
+class TableResult(CVResult, HtmlMixin, XlsxMixin):
     """TableResult"""
 
+    def __init__(self, data):
+        super().__init__(data)
+        HtmlMixin.__init__(self)
+        XlsxMixin.__init__(self)
+
     def save_to_html(self, save_path):
         if not save_path.lower().endswith(("html")):
             input_path = self["input_path"]