Change to not include any bindings that numpy already defines

gwastro · Oct 10, 2023 · 6f5deda · 6f5deda
1 parent 6311b4a
commit 6f5deda
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pycbc/io/record.py b/pycbc/io/record.py
@@ -46,11 +46,10 @@
 # =============================================================================
 #
 # add ligolw_types to numpy sctypeDict
-# but don't including float -> float32 and int -> int32 bindings, so as not
-# to change numpy defaults
+# but don't include bindings that numpy already defines
 numpy.sctypeDict.update({_k: _val
                          for (_k, _val) in ligolw_types.ToNumPyType.items()
-                         if _k != 'float' and _k != 'int'})
+                         if _k not in numpy.sctypeDict})
 
 # Annoyingly, numpy has no way to store NaNs in an integer field to indicate
 # the equivalent of None. This can be problematic for fields that store ids: