diff --git a/src/utils_data.py b/src/utils_data.py
index eea61c59be29003b17ae65f354064dd394628076..afa820cace1fa73ae6ee05e4c133494e191a108f 100644
--- a/src/utils_data.py
+++ b/src/utils_data.py
@@ -46,7 +46,6 @@ def create_label_dict(dataset : str, nn_model : str) -> dict:
     import sys
     import numpy as np
     import torchvision
-    from extra_keras_datasets import kmnist
     import torchvision.transforms as transforms
 
     transform = transforms.Compose(