summaryrefslogtreecommitdiffstats
path: root/python_modules/codegen.py
diff options
context:
space:
mode:
Diffstat (limited to 'python_modules/codegen.py')
-rw-r--r--python_modules/codegen.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/python_modules/codegen.py b/python_modules/codegen.py
index 55f513b..f324498 100644
--- a/python_modules/codegen.py
+++ b/python_modules/codegen.py
@@ -121,10 +121,10 @@ class CodeWriter:
def write(self, s):
# Ensure its a unicode string
- if six.PY2:
- s = unicode(s)
- else:
+ if six.PY3:
s = str(s)
+ else:
+ s = unicode(s)
if len(s) == 0:
return