From 7c33b33440dfa13141a266309075fc394ff7313e Mon Sep 17 00:00:00 2001
From: darkliang <11710911@mail.sustech.edu.cn>
Date: Tue, 18 Jan 2022 19:52:38 +0800
Subject: [PATCH] add input data check

---
 lib/data/dataset.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/lib/data/dataset.py b/lib/data/dataset.py
index 2ecd759..a06ed58 100644
--- a/lib/data/dataset.py
+++ b/lib/data/dataset.py
@@ -41,13 +41,14 @@ class DataSet(tordata.Dataset):
                 f.close()
             else:
                 raise ValueError('- Loader - just support .pkl !!!')
-            # if len(_) >= 200:
-            #     _ = _[:200]
             data_list.append(_)
-        for data in data_list:
+        for idx, data in enumerate(data_list):
             if len(data) != len(data_list[0]):
-                raise ValueError('Each input data should have the same length.')
-
+                raise ValueError(
+                    'Each input data({}) should have the same length.'.format(paths[idx]))
+            if len(data) == 0:
+                raise ValueError(
+                    'Each input data({}) should have at least one element.'.format(paths[idx]))
         return data_list
 
     def __getitem__(self, idx):
@@ -116,7 +117,8 @@ class DataSet(tordata.Dataset):
                                     seq_dirs, data_in_use) if use_bl]
                             seqs_info_list.append([*seq_info, seq_dirs])
                         else:
-                            msg_mgr.log_debug('Find no .pkl file in %s-%s-%s.'%(lab, typ, vie))
+                            msg_mgr.log_debug(
+                                'Find no .pkl file in %s-%s-%s.' % (lab, typ, vie))
             return seqs_info_list
 
         self.seqs_info = get_seqs_info_list(