Fix Kaldi stage_status num of args error (#308)

CoEDL · Sep 20, 2022 · a34a8ee · a34a8ee
1 parent b558851
commit a34a8ee
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/elpis/engines/common/objects/model.py b/elpis/engines/common/objects/model.py
@@ -59,6 +59,7 @@ def log(self, value: str):
 
     @stage_status.setter
     def stage_status(self, status_info: Tuple[str, str]):
+        # Caution: stage_status in the model accepts two args. Transcriber's stage_status accepts three.
         stage, status = status_info
         stage_status = self.config["stage_status"]
         stage_status[stage]["status"] = status

diff --git a/elpis/engines/common/objects/transcription.py b/elpis/engines/common/objects/transcription.py
@@ -49,6 +49,7 @@ def stage_status(self):
 
     @stage_status.setter
     def stage_status(self, vals: Tuple[str, str, str]):
+        # Caution: stage_status in the transcriber accepts three args. Model's stage_status only accepts two.
         stage, status, message = vals
         stage_status = self.config["stage_status"]
         stage_status[stage]["status"] = status

diff --git a/elpis/engines/kaldi/objects/model.py b/elpis/engines/kaldi/objects/model.py
@@ -263,7 +263,7 @@ def train():
                         print("stderr", error.stderr, file=file)
                         print("failed", file=file)
                     logger.error(f"Stage {stage} failed")
-                    self.stage_status = (stage, "failed", "", "LOG-C")
+                    self.stage_status = (stage, "failed")
                     break
 
         def run_training_in_background():