diff --git a/bert.py b/bert.py
index dfa6b83..8d1f0e3 100644
--- a/bert.py
+++ b/bert.py
@@ -1,5 +1,4 @@
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments, Trainer
-import random
 import torch
 
 PATHS = ['train/in.tsv', 'train/expected.tsv', 'dev-0/in.tsv', 'test-A/in.tsv', './dev-0/out.tsv', './test-A/out.tsv']