update

patch_madmom.py

@@ -5,6 +5,14 @@ from pathlib import Path
 
 FUTURE_LINE = "from __future__ import absolute_import, division, print_function"
 
+ALIASES = [
+    (re.compile(r"\bnp\.float\b"), "float"),
+    (re.compile(r"\bnp\.int\b"), "int"),
+    (re.compile(r"\bnp\.bool\b"), "bool"),
+    (re.compile(r"\bnp\.object\b"), "object"),
+    (re.compile(r"\bnp\.complex\b"), "complex"),
+]
+
 def find_madmom_root() -> Path:
     for p in site.getsitepackages():
         cand = Path(p) / "madmom"
@@ -25,28 +33,18 @@ def patch_processors(madmom_root: Path) -> None:
         proc.write_text(s2, encoding="utf-8")
 
 def patch_numpy_aliases(madmom_root: Path) -> None:
-    repls = [
-        ("np.float", "float"),
-        ("np.int", "int"),
-        ("np.bool", "bool"),
-        ("np.object", "object"),
-    ]
-
     patched = 0
     for f in madmom_root.rglob("*.py"):
         try:
             s = f.read_text(encoding="utf-8")
         except Exception:
             continue
-
         s2 = s
-        for a, b in repls:
-            s2 = s2.replace(a, b)
-
+        for pat, repl in ALIASES:
+            s2 = pat.sub(repl, s2)
         if s2 != s:
             f.write_text(s2, encoding="utf-8")
             patched += 1
-
     print(f"Patched NumPy aliases in {patched} madmom files")
 
 def patch_utils_future_safe(madmom_root: Path) -> None: