From 80fce623ec6502d59c4a2931c5a74ee3899be342 Mon Sep 17 00:00:00 2001
From: sjplimp <sjplimp@f3b2605a-c512-4ea7-a41b-209d697bcdaa>
Date: Fri, 15 Jul 2016 22:59:07 +0000
Subject: [PATCH] git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@15326
 f3b2605a-c512-4ea7-a41b-209d697bcdaa

---
 python/lammps.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/python/lammps.py b/python/lammps.py
index 114468e36e..bbc46cf2b9 100644
--- a/python/lammps.py
+++ b/python/lammps.py
@@ -79,6 +79,9 @@ class lammps:
         if cmdargs:
           cmdargs.insert(0,"lammps.py")
           narg = len(cmdargs)
+          for i in range(narg):
+            if type(cmdargs[i]) is str:
+              cmdargs[i] = cmdargs[i].encode()
           cargs = (c_char_p*narg)(*cmdargs)
           self.lib.lammps_open.argtypes = [c_int, c_char_p*narg, \
                                            MPI_Comm, c_void_p()]
@@ -98,6 +101,9 @@ class lammps:
         if cmdargs:
           cmdargs.insert(0,"lammps.py")
           narg = len(cmdargs)
+          for i in range(narg):
+            if type(cmdargs[i]) is str:
+              cmdargs[i] = cmdargs[i].encode()
           cargs = (c_char_p*narg)(*cmdargs)
           self.lmp = c_void_p()
           self.lib.lammps_open_no_mpi(narg,cargs,byref(self.lmp))
@@ -211,12 +217,12 @@ class lammps:
       return result
     if type == 1:
       self.lib.lammps_extract_global.restype = POINTER(c_int)
-      nlocalptr = self.lib.lammps_extract_global(self.lmp,"nlocal")
+      nlocalptr = self.lib.lammps_extract_global(self.lmp,"nlocal".encode())
       nlocal = nlocalptr[0]
       result = (c_double*nlocal)()
       self.lib.lammps_extract_variable.restype = POINTER(c_double)
       ptr = self.lib.lammps_extract_variable(self.lmp,name,group)
-      for i in xrange(nlocal): result[i] = ptr[i]
+      for i in range(nlocal): result[i] = ptr[i]
       self.lib.lammps_free(ptr)
       return result
     return None