work around python 2 bug

desihub · Apr 12, 2017 · ae61fdd · ae61fdd
1 parent 463b933
commit ae61fdd
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/py/desiutil/census.py b/py/desiutil/census.py
@@ -422,9 +422,14 @@ def output_csv(summary, filename):
             row.append(str(number[d][y]))
             row.append(str(size[d][y]))
         data.append(row)
-    with open(filename, 'w', newline='') as csvfile:
-        writer = csv.writer(csvfile)
-        writer.writerows(data)
+    try:
+        csvfile = open(filename, 'w', newline='')
+    except TypeError:
+        # Python 2
+        csvfile = open(filename, 'w')
+    writer = csv.writer(csvfile)
+    writer.writerows(data)
+    csvfile.close()
     return data