fix: address PR #32 review feedback

- Regenerate uv.lock (was stale: 0.5.7 vs 0.5.8 in pyproject.toml).
  Pin ruff to >=0.15.22,<0.16 to avoid 0.16 upgrade introducing 55
  new pre-existing lint errors (RUF022/RUF059/I001/etc.); those fixes
  belong in a separate PR.
- Add --base-channels to train_audio_diffusion.py and forward to the
  fallback BaselineAudioDecoder (was hardcoded to 64, inconsistent
  with the docs table and the decoder script).
- Wrap subprocess in test_script_cli.py with pytest.fail for clean
  error output on import failure (was opaque CalledProcessError).
- Add missing pytest import in test_script_cli.py.

Entire-Checkpoint: 0c2d99a5ab77

Author: Mec-iS

pyproject.toml

@@ -13,7 +13,7 @@ dependencies = [
     "numpy>=2.5.1",
     "pedalboard>=0.9.0",
     "pytest>=9.1.1",
-    "ruff>=0.15.22",
+    "ruff>=0.15.22,<0.16",
     "scipy>=1.14.0",
     "soundfile>=0.13.1",
     "structlog>=26.1.0",

scripts/train_audio_diffusion.py

@@ -34,6 +34,13 @@ def main() -> None:
     parser.add_argument("--depth", type=int, default=4)
     parser.add_argument("--num-heads", type=int, default=4)
     parser.add_argument("--q", type=int, default=8)
+    parser.add_argument(
+        "--base-channels",
+        type=int,
+        default=64,
+        help="Base channels for the fallback baseline decoder (only used when "
+        "--decoder is not provided)",
+    )
     parser.add_argument("--batch-size", type=int, default=4)
     parser.add_argument("--epochs", type=int, default=50)
     parser.add_argument("--lr", type=float, default=1e-4)
@@ -87,10 +94,13 @@ def extract_features(self, x):
     else:
         encoder = EnCodecEncoder()
 
+    # Fallback baseline decoder (only used when --decoder is not provided;
+    # the DiT trains against the frozen encoder's latents, so this decoder
+    # is just a placeholder for the AudioVAE wrapper).
     decoder = BaselineAudioDecoder(
         latent_channels=args.latent_channels,
         out_channels=1,
-        base_channels=64,
+        base_channels=args.base_channels,
     )
     if args.decoder and Path(args.decoder).exists():
         decoder.load_state_dict(torch.load(args.decoder, weights_only=False))

tests/test_script_cli.py

@@ -14,18 +14,23 @@
 import sys
 from pathlib import Path
 
+import pytest
+
 SCRIPTS_DIR = Path(__file__).resolve().parent.parent / "scripts"
 
 
 def _help_output(script: str) -> str:
     """Run a script with --help and return its stdout."""
-    result = subprocess.run(
-        [sys.executable, str(SCRIPTS_DIR / script), "--help"],
-        capture_output=True,
-        text=True,
-        check=True,
-        timeout=30,
-    )
+    try:
+        result = subprocess.run(
+            [sys.executable, str(SCRIPTS_DIR / script), "--help"],
+            capture_output=True,
+            text=True,
+            check=True,
+            timeout=30,
+        )
+    except subprocess.CalledProcessError as e:
+        pytest.fail(f"{script} --help failed:\n{e.stderr}")
     return result.stdout