diff --git a/program/tool-prepare-dataset/prepare.py b/program/tool-prepare-dataset/prepare.py index f6fcf3f..5a6cfdd 100755 --- a/program/tool-prepare-dataset/prepare.py +++ b/program/tool-prepare-dataset/prepare.py @@ -214,11 +214,15 @@ def prepare_reshape(row): dataset_files = [] desc_dataset_files = {} for row in rows: - # Column 0 defines network name but not all the rows has it - if row[0]: - net_name = row[0] - else: - row[0] = net_name + # Column 0 defines network name but not all rows have it + try: + if row[0]: + net_name = row[0] + else: + row[0] = net_name + except: + ck.out('Skip row {}'.format(row)) + continue # Run preparation func try: