diff --git a/egs/yesno/ASR/tdnn/train.py b/egs/yesno/ASR/tdnn/train.py
index e476e8beb..f2463428b 100755
--- a/egs/yesno/ASR/tdnn/train.py
+++ b/egs/yesno/ASR/tdnn/train.py
@@ -4,11 +4,10 @@ import argparse
 import logging
 from pathlib import Path
 from shutil import copyfile
-from typing import Optional, Tuple, List
+from typing import Optional, Tuple
 
 import k2
 import torch
-import torch.distributed as dist
 import torch.multiprocessing as mp
 import torch.nn as nn
 import torch.optim as optim
@@ -414,8 +413,8 @@ def train_one_epoch(
         if batch_idx % params.log_interval == 0:
             logging.info(
                 f"Epoch {params.cur_epoch}, batch {batch_idx}, loss[{loss_info}]"
-                f"tot_loss[{tot_loss}], batch size: {batch_size}"
-            )
+                f"tot_loss[{tot_loss}], batch size: {batch_size}")
+            
         if batch_idx % 10 == 0:
 
             if tb_writer is not None: