diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py
index fe265eb81..018e2827b 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py
@@ -17,13 +17,12 @@
 
 import collections
 from itertools import repeat
-from typing import List, Optional, Tuple
+from typing import Optional, Tuple
 
-import torch
-import torch.nn as nn
 from torch import Tensor, _VF
-
+import torch
 import torch.backends.cudnn.rnn as rnn
+import torch.nn as nn
 
 
 def _ntuple(n):