changed error check and type

lbl-srg · Apr 12, 2024 · 27d7fd1 · 27d7fd1
1 parent 64cda69
commit 27d7fd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyfunnel/pyfunnel.py b/pyfunnel/pyfunnel.py
@@ -89,8 +89,8 @@
         data[s] = dict(x=[], y=[])
         with open(vars(args)[s]) as csvfile:
             spamreader = csv.reader(csvfile)
-            if (len(next(spamreader)) != 2):
-                raise RuntimeError("The {} CSV file must have exactly two columns.".format(s))
+            if (len(spamreader[0]) != 2):
+                raise IOError("The {} CSV file must have exactly two columns.".format(s))
             for row in spamreader:
                 try:
                     data[s]['x'].append(float(row[0]))