diff --git a/function.py b/function.py
index 4b5f68a..3eddcea 100644
--- a/function.py
+++ b/function.py
@@ -1,6 +1,7 @@
 # function.py
 import numpy as np
 import random
+import sklearn
 from sklearn.preprocessing import StandardScaler
 
 """