Fix writing metadata when no lm is trained

1 files changed, 2 insertions, 0 deletions

diff --git a/bicleaner_ai/bicleaner_ai_train.py b/bicleaner_ai/bicleaner_ai_train.py
index c0bb7a4..2542ea0 100755
--- a/bicleaner_ai/bicleaner_ai_train.py
+++ b/bicleaner_ai/bicleaner_ai_train.py
@@ -197,6 +197,8 @@ def perform_training(args):
         args.parallel_train.seek(0)
         args.input = args.parallel_train
         lm_stats = train_fluency_filter(args)
+    else:
+        lm_stats = None
     args.parallel_train.close()
     args.parallel_valid.close()