patch madmom

Dockerfile

@@ -25,6 +25,7 @@ RUN pip install --no-cache-dir --no-build-isolation madmom
 RUN python - <<'PY'
 import site, pathlib, re
 
+# locate madmom package
 pkg_root = None
 for p in site.getsitepackages():
     cand = pathlib.Path(p) / "madmom"
@@ -34,32 +35,75 @@ for p in site.getsitepackages():
 if pkg_root is None:
     raise SystemExit("madmom package not found")
 
-repls = [
-    ("from collections import MutableSequence", "from collections.abc import MutableSequence"),
-    ("np.float", "float"),   # or "np.float64"
-    ("np.int", "int"),
-    ("np.bool", "bool"),
-    ("np.object", "object"),
-]
-
-patched = 0
-for f in pkg_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)
+# 1) Fix processors.py: MutableSequence import for py3.10+
+proc = pkg_root / "processors.py"
+if proc.exists():
+    s = proc.read_text(encoding="utf-8")
+    s2 = s.replace("from collections import MutableSequence",
+                   "from collections.abc import MutableSequence")
     if s2 != s:
-        f.write_text(s2, encoding="utf-8")
-        patched += 1
+        proc.write_text(s2, encoding="utf-8")
+        print("Patched", proc)
+
+# 2) Fix numpy aliases if present (np.float etc.)
+for f in [pkg_root / "io" / "__init__.py"]:
+    if f.exists():
+        s = f.read_text(encoding="utf-8")
+        s2 = (s.replace("np.float", "float")
+               .replace("np.int", "int")
+               .replace("np.bool", "bool")
+               .replace("np.object", "object"))
+        if s2 != s:
+            f.write_text(s2, encoding="utf-8")
+            print("Patched", f)
+
+# 3) Fix Python2 type names in madmom/utils/__init__.py
+u = pkg_root / "utils" / "__init__.py"
+if not u.exists():
+    raise SystemExit("madmom/utils/__init__.py not found")
+
+s = u.read_text(encoding="utf-8")
+
+# Inject compatibility definitions near the top (after imports)
+# We'll add them right after the first import block.
+lines = s.splitlines(True)
+insert_at = 0
+for i, line in enumerate(lines):
+    # after the last consecutive import line at the top
+    if line.startswith("import ") or line.startswith("from "):
+        insert_at = i + 1
+    elif i > 0:
+        break
+
+compat = """
+# --- Compatibility for Python 3 ---
+try:
+    basestring
+except NameError:
+    basestring = str
+
+try:
+    integer
+except NameError:
+    integer = int
+# ---------------------------------
+"""
+
+# Only insert if not already present
+if "basestring = str" not in s:
+    lines.insert(insert_at, compat)
+    s = "".join(lines)
+
+# Ensure string_types / integer_types are well-defined even if old code is weird
+s = re.sub(r"string_types\s*=\s*basestring", "string_types = (str,)", s)
+s = re.sub(r"integer_types\s*=\s*\(int,\s*integer\)", "integer_types = (int,)", s)
 
-print(f"Patched {patched} madmom files under {pkg_root}")
+u.write_text(s, encoding="utf-8")
+print("Patched", u)
 PY
 
 
-RUN python -c "import setuptools, pkg_resources; print('setuptools OK')"
+RUN python -c "import madmom; print('madmom import OK')"
 
 # Copy the rest of the repo
 COPY . /app