Fix lazypredict import

swiss-ai-center · Mar 27, 2024 · 84d27e4 · 84d27e4
1 parent 739679f
commit 84d27e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -24,7 +24,7 @@
 
 # Imports required by the service's model
 import pandas as pd
-from lazypredict.Supervised import LazyRegressor
+from lazypredict import LazyRegressor
 from sklearn.model_selection import train_test_split
 import io