4 splits

1 files changed, 2 insertions, 1 deletions

diff --git a/split-csv.py b/split-csv.py
index e7c12883..122d2ddc 100644
--- a/split-csv.py
+++ b/split-csv.py
@@ -4,6 +4,7 @@ import csv
 from math import ceil
 import subprocess
 import random
+from util import *
 
 import click
 
@@ -16,7 +17,7 @@ def split_csv(count, has_keys, shuffle, filename):
   """Split a CSV into groups."""
   with open(filename, 'r') as f:
     reader = csv.reader(f)
-    lines = list(reader)
+    lines = list(unfussy_reader(reader))
     if has_keys:
       keys = lines[0]
       lines = lines[1:]