import sklearn.model_selection

RichieHakim · Sep 27, 2024 · be01585 · be01585
1 parent d589a70
commit be01585
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/bnpm/automatic_regression.py b/bnpm/automatic_regression.py
@@ -7,6 +7,7 @@
 import numpy as np
 import torch
 import sklearn
+import sklearn.model_selection
 import optuna
 
 from . import path_helpers

diff --git a/bnpm/cross_validation.py b/bnpm/cross_validation.py
@@ -1,6 +1,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import sklearn
+import sklearn.model_selection
 
 def make_cv_indices(
     cv,