Fix for #281: string literal auto-encoded with 'ascii' for Py3

pydata · Sep 12, 2017 · a2d3efe · a2d3efe
1 parent bece07a
commit a2d3efe
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/RELEASE_NOTES.rst b/RELEASE_NOTES.rst
@@ -20,6 +20,9 @@ Changes from 2.6.2 to 2.6.3
 - Fix for #277 whereby arrays of shape (1,...) would be reduced as 
   if they were full reduction. Behavoir now matches that of NumPy.
 
+- String literals are automatically encoded into 'ascii' bytes for 
+  convience (see #281).
+
 Changes from 2.6.1 to 2.6.2
 ===========================
 

diff --git a/numexpr/necompiler.py b/numexpr/necompiler.py
@@ -334,6 +334,8 @@ def convertConstantToKind(x, kind):
     # Exception for 'float' types that will return the NumPy float32 type
     if kind == 'float':
         return numpy.float32(x)
+    elif sys.version_info[0] >= 3 and isinstance(x,str):
+        return x.encode('ascii')
     return kind_to_type[kind](x)