Fix python3 error

xianqiangsun · Jan 2, 2014 · cc03ef8 · cc03ef8
1 parent b1dcae8
commit cc03ef8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/MDTraj/lh5.py b/MDTraj/lh5.py
@@ -1,4 +1,4 @@
-#
+##############################################################################
 # MDTraj: A Python Library for Loading, Saving, and Manipulating
 #         Molecular Dynamics Trajectories.
 # Copyright 2012-2014 Stanford University and the Authors
@@ -18,7 +18,7 @@
 #
 # You should have received a copy of the GNU Lesser General Public
 # License along with MDTraj. If not, see <http://www.gnu.org/licenses/>.
-#
+##############################################################################
 
 """ MSMBuilder2 "LH5" trajectory format.
 """
@@ -167,7 +167,8 @@ def topology(self):
         topology : mdtraj.Topology
             A topology object
         """
-        if np.all(self._handle.root.AtomID[:] == 0) and np.all(self._handle.root.AtomNames[:] == ''):
+        if np.all(self._handle.root.AtomID[:] == 0) and (np.all(self._handle.root.AtomNames[:] == b'')
+            or np.all(self._handle.root.AtomNames[:] == u'')):
             return None
 
         return _topology_from_arrays(