From 437c25891d7c8e2101b8bff9a72f8f9a866069d9 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 6 Nov 2018 15:25:50 +0100
Subject: 4 splits

---
 split-csv.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'split-csv.py')

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:]
-- 
cgit v1.2.3-70-g09d2