batchalign 0.8.2.post3__tar.gz → 0.8.2.post4__tar.gz

{batchalign-0.8.2.post3/batchalign.egg-info → batchalign-0.8.2.post4}/PKG-INFO

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

{batchalign-0.8.2.post3 → batchalign-0.8.2.post4}/batchalign/cli/dispatch.py

@@ -220,7 +220,7 @@ def _run_pipeline_for_file(command, pipeline, file, output, loader_info, writer_
         main_doc = CHATFile(path=str(p)).doc
         gold_doc = CHATFile(path=str(gold_path), special_mor_=True).doc
 
-        # Pipeline: morphosyntax(main) -> compare -> compare_analysis
+        # Pipeline: compare (runs morphosyntax internally) -> compare_analysis
         result = pipeline(main_doc, callback=progress_callback, gold=gold_doc)
 
         # Write annotated CHAT
@@ -388,7 +388,7 @@ Cmd2Task = {
     "coref": "coref",
     "translate": "translate",
     "opensmile": "opensmile",
-    "compare": "morphosyntax,compare,compare_analysis",
+    "compare": "compare,compare_analysis",
 }
 
 # this is the main runner used by all functions

{batchalign-0.8.2.post3 → batchalign-0.8.2.post4}/batchalign/pipelines/analysis/compare.py

@@ -4,7 +4,7 @@ Engines for transcript comparison against gold-standard references.
 
 CompareEngine (PROCESSING): Aligns main vs gold transcripts word-by-word
 using the same conform/match_fn logic as WER evaluation, then annotates
-each main utterance with comparison tokens (%xsrep / %xsmor).
+each gold utterance with comparison tokens (%xsrep / %xsmor).
 
 CompareAnalysisEngine (ANALYSIS): Reads the comparison annotations and
 computes error-rate metrics for CSV output.
@@ -208,6 +208,11 @@ def _find_best_segment(gold_tokens, main_tokens, mfn):
                 elif len_delta == best_len_delta and end > best[1]:
                     best = (start, end)
 
+    # If no tokens overlap at all, return an empty window so the caller
+    # doesn't consume main tokens that belong to a later gold utterance.
+    if best_score <= 0:
+        return 0, 0
+
     return best
 
 
@@ -249,6 +254,15 @@ def conform_with_mapping(words, conform_fn):
 class CompareEngine(BatchalignEngine):
     tasks = [Task.COMPARE]
 
+    def __init__(self):
+        self._stanza = None
+
+    def _get_stanza(self):
+        if self._stanza is None:
+            from batchalign.pipelines.morphosyntax import StanzaEngine
+            self._stanza = StanzaEngine()
+        return self._stanza
+
     def process(self, doc, **kwargs):
         gold = kwargs.get("gold")
         if not gold or not isinstance(gold, Document):
@@ -256,6 +270,12 @@ class CompareEngine(BatchalignEngine):
                 f"CompareEngine requires a 'gold' Document kwarg, got '{type(gold)}'"
             )
 
+        # --- 0. Run morphosyntax on both docs ---
+        stanza = self._get_stanza()
+        non_gold_kwargs = {k: v for k, v in kwargs.items() if k != "gold"}
+        doc = stanza.process(doc, **non_gold_kwargs)
+        gold = stanza.process(gold, **non_gold_kwargs)
+
         # --- 1. Extract words from main utterances ---
         main_utterances = [
             u for u in doc.content if isinstance(u, Utterance)
@@ -343,16 +363,14 @@ class CompareEngine(BatchalignEngine):
                     gold_form = gold_info[orig_gold_idx][2]
                     last_gold_form_idx = gold_form_idx
 
+                    # Copy timing from main; keep gold's own morphology
                     if main_form.time is not None:
                         gold_form.time = main_form.time
-                    if main_form.morphology is not None:
-                        gold_form.morphology = main_form.morphology
-                    if main_form.dependency is not None:
-                        gold_form.dependency = main_form.dependency
+                    gold_form.dependency = None
 
                     utt_positioned[utt_idx].append((gold_form_idx, CompareToken(
                         text=item.key,
-                        pos=_get_pos(main_form),
+                        pos=_get_pos(gold_form),
                         status="match"
                     )))
                     local_main_cursor += 1
@@ -365,6 +383,8 @@ class CompareEngine(BatchalignEngine):
                         gold_form = gold_info[orig_gold_idx][2]
                         last_gold_form_idx = gold_form_idx
 
+                        gold_form.dependency = None
+
                         utt_positioned[utt_idx].append((gold_form_idx, CompareToken(
                             text=item.key,
                             pos=_get_pos(gold_form),
@@ -401,6 +421,7 @@ class CompareEngine(BatchalignEngine):
                     form.morphology = [Morphology(
                         pos="PUNCT", lemma=form.text.strip(), feats=""
                     )]
+                form.dependency = None
             utt_positioned[utt_idx].sort(key=lambda x: x[0])
 
         # --- 7. Set comparison on each gold utterance ---

batchalign-0.8.2.post4/batchalign/version

@@ -0,0 +1,3 @@
+0.8.2-post.4
+March 30 2026
+More compare bug fixes

{batchalign-0.8.2.post3 → batchalign-0.8.2.post4/batchalign.egg-info}/PKG-INFO

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

batchalign-0.8.2.post3/batchalign/version

@@ -1,3 +0,0 @@
-0.8.2-post.3
-March 28 2026
-fix bugs