diff --git a/tests/test_training/train_CI.py b/tests/test_training/train_CI.py
index 481b4c8..b05772b 100644
--- a/tests/test_training/train_CI.py
+++ b/tests/test_training/train_CI.py
@@ -2,7 +2,7 @@
 # -*- encoding: utf-8 -*-
 import os
 
-# pylint: disable=C0413,W0612,W0611,E402
+# pylint: disable=C0413,W0612,W0611
 import socket
 import sys
 import time