Misc fixes

montreal_forced_aligner/abc.py

@@ -426,7 +426,9 @@ def parse_args(cls, args: Optional[Namespace], unknown_args: Optional[List[str]]
                     val = unknown_args[i + 1]
                 unknown_dict[name] = val
         for name, param_type in param_types.items():
-            if name.endswith("_directory") or name.endswith("_path"):
+            if (name.endswith("_directory") and name != "audio_directory") or name.endswith(
+                "_path"
+            ):
                 continue
             if args is not None and hasattr(args, name):
                 params[name] = param_type(getattr(args, name))

montreal_forced_aligner/acoustic_modeling/trainer.py

@@ -346,6 +346,7 @@ def align(self) -> None:
                     f"Analyzing alignment diagnostics for {self.current_aligner.identifier} on the full corpus"
                 )
             self.compile_information()
+            self.collect_alignments()
             with open(done_path, "w"):
                 pass
         except Exception as e: