Huiran Yu commited on
Commit
a069413
·
1 Parent(s): 4e3ad8c

patch all files

Browse files
Files changed (1) hide show
  1. patch_madmom.py +23 -13
patch_madmom.py CHANGED
@@ -25,19 +25,29 @@ def patch_processors(madmom_root: Path) -> None:
25
  proc.write_text(s2, encoding="utf-8")
26
 
27
  def patch_numpy_aliases(madmom_root: Path) -> None:
28
- # Patch the known failing location; you can broaden to rglob if needed.
29
- f = madmom_root / "io" / "__init__.py"
30
- if not f.exists():
31
- return
32
- s = f.read_text(encoding="utf-8")
33
- s2 = (
34
- s.replace("np.float", "float")
35
- .replace("np.int", "int")
36
- .replace("np.bool", "bool")
37
- .replace("np.object", "object")
38
- )
39
- if s2 != s:
40
- f.write_text(s2, encoding="utf-8")
 
 
 
 
 
 
 
 
 
 
41
 
42
  def patch_utils_future_safe(madmom_root: Path) -> None:
43
  u = madmom_root / "utils" / "__init__.py"
 
25
  proc.write_text(s2, encoding="utf-8")
26
 
27
  def patch_numpy_aliases(madmom_root: Path) -> None:
28
+ repls = [
29
+ ("np.float", "float"),
30
+ ("np.int", "int"),
31
+ ("np.bool", "bool"),
32
+ ("np.object", "object"),
33
+ ]
34
+
35
+ patched = 0
36
+ for f in madmom_root.rglob("*.py"):
37
+ try:
38
+ s = f.read_text(encoding="utf-8")
39
+ except Exception:
40
+ continue
41
+
42
+ s2 = s
43
+ for a, b in repls:
44
+ s2 = s2.replace(a, b)
45
+
46
+ if s2 != s:
47
+ f.write_text(s2, encoding="utf-8")
48
+ patched += 1
49
+
50
+ print(f"Patched NumPy aliases in {patched} madmom files")
51
 
52
  def patch_utils_future_safe(madmom_root: Path) -> None:
53
  u = madmom_root / "utils" / "__init__.py"