Ensure approximant field is correct format (#4646)

gwastro · Feb 20, 2024 · aa05af6 · aa05af6
1 parent e7e9666
commit aa05af6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/bin/pycbc_convertinjfiletohdf b/bin/pycbc_convertinjfiletohdf
@@ -151,7 +151,11 @@ class LVKNewStyleInjectionSet(object):
             elif lvk_name == 't_co_gps_add':
                 continue
             else:
-                data[pycbc_name] = self.inj_file[f'{self.subdir}/{lvk_name}'][:]
+                lvk_file_dset = self.inj_file[f'{self.subdir}/{lvk_name}']
+                if lvk_file_dset.dtype.char in ['U', 'O']:
+                    data[pycbc_name] = lvk_file_dset[:].astype('S')
+                else:
+                    data[pycbc_name] = lvk_file_dset[:]
         return data
 
 
@@ -187,4 +191,4 @@ if injclass is not None:
     CBCHDFInjectionSet.write(args.output_file, samples)
 else:
     # Assume the injection is a HDF file (PyCBC format), only make a copy
-    shutil.copy(args.injection_file, args.output_file)
+    shutil.copy(args.injection_file, args.output_file)