diff --git a/embedding_converter/src/dataset.py b/embedding_converter/src/dataset.py
index 6ed1a4f..060dee5 100644
--- a/embedding_converter/src/dataset.py
+++ b/embedding_converter/src/dataset.py
@@ -1,5 +1,4 @@
 import glob
-import random
 
 import cv2
 from torch import Tensor