Merge pull request #208 from HERA-Team/allow_force_pspeccontainer

Handle bytes/str comparisons in PSpecContainer

hera_pspec/container.py

@@ -81,15 +81,19 @@ def _load_pspec(self, pspec_group):
         pspec_group : HDF5 group
             Group containing datasets that contain power spectrum and
             supporting information, in a standard format expected by UVPSpec.
-
+        
         Returns
         -------
         uvp : UVPSpec
             Returns a UVPSpec object constructed from the input HDF5 group.
         """
         # Check that group is tagged as containing UVPSpec (pspec_type attribute)
         if 'pspec_type' in list(pspec_group.attrs.keys()):
-            if pspec_group.attrs['pspec_type'] != uvpspec.UVPSpec.__name__:
+
+            # Convert bytes -> str if needed
+            pspec_type = pspec_group.attrs['pspec_type']
+            if isinstance(pspec_type, bytes): pspec_type = pspec_type.decode()
+            if pspec_type != uvpspec.UVPSpec.__name__:
                 raise TypeError("HDF5 group is not tagged as a UVPSpec object.")
         else:
             raise TypeError("HDF5 group is not tagged as a UVPSpec object.")
@@ -206,7 +210,7 @@ def get_pspec(self, group, psname=None):
 
         psname : str, optional
             The name of the power spectrum to return.
-
+        
         Returns
         -------
         uvp : UVPSpec or list of UVPSpec