Merge pull request #2284 from AllenInstitute/fix_py36_compare_bytes

Fixes issue in which py36 returns string and cannot compare to bytes
AllenInstitute · Jan 3, 2022 · 656e01a · 656e01a
2 parents 0a924dc + cbb2396
commit 656e01a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/allensdk/brain_observatory/ecephys/ecephys_session_api/ecephys_nwb1_session_api.py b/allensdk/brain_observatory/ecephys/ecephys_session_api/ecephys_nwb1_session_api.py
@@ -62,8 +62,10 @@ def __init__(self, path, *args, **kwargs):
         try:
             # check file is a valid NWB 1 file
             version_str = self._h5_root['nwb_version'][()]
-            if not (version_str.startswith(b'NWB-1.') or
-                    version_str.startswith(b'1.')):
+            if isinstance(version_str, bytes):
+                version_str = version_str.decode('utf-8')
+            if not (version_str.startswith('NWB-1.') or
+                    version_str.startswith('1.')):
                 raise Exception(
                     '{} is not a valid NWB 1 file path'.format(self._path))
         except Exception: