Skip weird rows (in particular, empty ones).

ctuning · Dec 4, 2018 · dfe75c4 · dfe75c4
1 parent 73a058e
commit dfe75c4
Showing 1 changed file with 9 additions and 5 deletions.
diff --git 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: