From ef7116d77f05751c59de1781ee34c4e3e7dfdb0b Mon Sep 17 00:00:00 2001
From: Liang Yang <mail.liangyang@gmail.com>
Date: Wed, 30 Mar 2022 17:09:09 +0800
Subject: [PATCH] Fix tabular printing of dataset information (#53)

Co-authored-by: linyiqi <linyq29@gmail.com>
---
 mmfewshot/detection/datasets/base.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mmfewshot/detection/datasets/base.py b/mmfewshot/detection/datasets/base.py
index b1d69ac..4b6dfdd 100644
--- a/mmfewshot/detection/datasets/base.py
+++ b/mmfewshot/detection/datasets/base.py
@@ -539,6 +539,8 @@ class BaseFewShotDataset(CustomDataset):
             if len(row_data) == 10:
                 table_data.append(row_data)
                 row_data = []
+        if len(row_data) != 0:
+            table_data.append(row_data)
 
         table = AsciiTable(table_data)
         result += table.table