diff --git a/parakeet/training/chekpoint.py b/parakeet/training/chekpoint.py
index 0cfdd5b..330834a 100644
--- a/parakeet/training/chekpoint.py
+++ b/parakeet/training/chekpoint.py
@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import operator
-from queue import PriorityQueue
 from typing import Callable, Mapping
 from pathlib import Path
 
@@ -42,7 +40,7 @@ class KBest(object):
     >>> import shutil
     >>> import paddle
     >>> from paddle import nn
-     
+    
     >>> model = nn.Linear(2, 3)
     >>> def save_model(path):
     ...     paddle.save(model.state_dict())