diff --git a/cyber-foundation/training-inference/prepare-prefix.py b/cyber-foundation/training-inference/prepare-prefix.py
index 7b12f37..e10bf59 100644
--- a/cyber-foundation/training-inference/prepare-prefix.py
+++ b/cyber-foundation/training-inference/prepare-prefix.py
@@ -39,7 +39,6 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-import json
 import os
 import pickle
 
diff --git a/cyber-foundation/training-inference/train-prefix.py b/cyber-foundation/training-inference/train-prefix.py
index 3506c49..425fcb5 100644
--- a/cyber-foundation/training-inference/train-prefix.py
+++ b/cyber-foundation/training-inference/train-prefix.py
@@ -53,8 +53,6 @@
 from torch.distributed import init_process_group
 from torch.nn.parallel import DistributedDataParallel as DDP
 
-
-
 out_dir = '../model/'
 eval_interval = 100
 log_interval = 1