py 2/3 compatibility for setup

nipy · Sep 8, 2016 · 0f9a620 · 0f9a620
1 parent 1a9bef0
commit 0f9a620
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -98,7 +98,10 @@ def run(self):
             cfg_parser.read(pjoin(pkg_dir, 'COMMIT_INFO.txt'))
             cfg_parser.set('commit hash', 'install_hash', repo_commit)
             out_pth = pjoin(self.build_lib, pkg_dir, 'COMMIT_INFO.txt')
-            cfg_parser.write(open(out_pth, 'wt'))
+            if PY3:
+                cfg_parser.write(open(out_pth, 'wt'))
+            else:
+                cfg_parser.write(open(out_pth, 'wb'))
     return MyBuildPy