batchalign 0.7.22.post4__tar.gz → 0.7.22.post7__tar.gz

{batchalign-0.7.22.post4/batchalign.egg-info → batchalign-0.7.22.post7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: batchalign
-Version: 0.7.22.post4
+Version: 0.7.22.post7
 Summary: Python Speech Language Sample Analysis
 Author: Brian MacWhinney, Houjun Liu
 Author-email: macw@cmu.edu, houjun@cmu.edu

{batchalign-0.7.22.post4 → batchalign-0.7.22.post7}/batchalign/cli/cli.py

@@ -179,7 +179,7 @@ def transcribe(ctx, in_dir, out_dir, lang, num_speakers, **kwargs):
 
     def writer(doc, output):
         doc.content.insert(0, CustomLine(id="Comment", type=CustomLineType.INDEPENDENT,
-                                         content=f"Batchalign {VERSION_NUMBER.strip()}, ASR Engine {asr}. Unchecked output of ASR model; do not use."))
+                                         content=f"Batchalign {VERSION_NUMBER.strip()}, ASR Engine {asr}. Unchecked output of ASR model."))
         CHATFile(doc=doc).write(output
                                 .replace(".wav", ".cha")
                                 .replace(".WAV", ".cha")

{batchalign-0.7.22.post4 → batchalign-0.7.22.post7}/batchalign/pipelines/analysis/eval.py

@@ -25,7 +25,7 @@ def conform(x):
     result = []
     for i in x:
         if i.strip().lower() in joined_compounds:
-            for k in compounds[joined_compounds.index(i.strip())]:
+            for k in compounds[joined_compounds.index(i.strip().lower())]:
                 result.append(k)
         elif i.strip() in lowered_abbrev:
             for j in i.strip():

{batchalign-0.7.22.post4 → batchalign-0.7.22.post7}/batchalign/pipelines/avqi/engine.py

@@ -11,6 +11,7 @@ from typing import Tuple, Dict, Optional
 import os
 from pathlib import Path
 import logging
+import torchaudio
 
 from batchalign.pipelines.base import BatchalignEngine
 from batchalign.document import Task, Document
@@ -246,8 +247,37 @@ class AVQIEngine(BatchalignEngine):
         
         L.info(f"Calculating AVQI for CS: {cs_path.name}, SV: {sv_name}")
 
+        # Create mono versions of both files
+        cs_mono_path = None
+        sv_mono_path = None
+
         try:
-            avqi_score, features = self.calculate_avqi_features(str(cs_file), str(sv_file))
+            # Convert cs_file to mono
+            L.info(f"Converting {cs_path.name} to mono")
+            cs_waveform, cs_sample_rate = torchaudio.load(str(cs_file))
+            if cs_waveform.shape[0] > 1:
+                cs_mono = cs_waveform.mean(dim=0, keepdim=True)
+            else:
+                cs_mono = cs_waveform
+
+            # Create mono filename: file_name.cs.[extension].mono
+            cs_mono_path = cs_path.parent / f"{cs_path.name}.mono.wav"
+            torchaudio.save(str(cs_mono_path), cs_mono, cs_sample_rate)
+
+            # Convert sv_file to mono
+            L.info(f"Converting {sv_name} to mono")
+            sv_waveform, sv_sample_rate = torchaudio.load(str(sv_file))
+            if sv_waveform.shape[0] > 1:
+                sv_mono = sv_waveform.mean(dim=0, keepdim=True)
+            else:
+                sv_mono = sv_waveform
+
+            # Create mono filename: file_name.sv.[extension].mono
+            sv_mono_path = sv_file.parent / f"{sv_file.name}.mono.wav"
+            torchaudio.save(str(sv_mono_path), sv_mono, sv_sample_rate)
+
+            # Calculate AVQI using mono versions
+            avqi_score, features = self.calculate_avqi_features(str(cs_mono_path), str(sv_mono_path))
 
             results = {
                 'avqi': avqi_score,
@@ -277,4 +307,12 @@ class AVQIEngine(BatchalignEngine):
                 'tilt': 0.0,
                 'error': str(e),
                 'success': False
-            }
+            }
+        finally:
+            # Clean up temporary mono files
+            if cs_mono_path and cs_mono_path.exists():
+                L.info(f"Cleaning up temporary file: {cs_mono_path.name}")
+                cs_mono_path.unlink()
+            if sv_mono_path and sv_mono_path.exists():
+                L.info(f"Cleaning up temporary file: {sv_mono_path.name}")
+                sv_mono_path.unlink()

batchalign-0.7.22.post7/batchalign/version

@@ -0,0 +1,3 @@
+0.7.22-post.7
+October 21st, 2025
+Patch two small bugs

{batchalign-0.7.22.post4 → batchalign-0.7.22.post7/batchalign.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: batchalign
-Version: 0.7.22.post4
+Version: 0.7.22.post7
 Summary: Python Speech Language Sample Analysis
 Author: Brian MacWhinney, Houjun Liu
 Author-email: macw@cmu.edu, houjun@cmu.edu

batchalign-0.7.22.post4/batchalign/version

@@ -1,3 +0,0 @@
-0.7.22-post.4
-October 17th, 2025
-Bump certs