diff --git a/data/data_gen.py b/data/data_gen.py
index 773500c..a581494 100644
--- a/data/data_gen.py
+++ b/data/data_gen.py
@@ -7,7 +7,6 @@
 
 import numpy as np
 np.random.seed(42)
-import pandas as pd
 from sklearn.datasets import load_boston, load_iris
 
 def main():