From be01585b781745d93703e69575306af1c1851863 Mon Sep 17 00:00:00 2001
From: RichieHakim <RichHakim@gmail.com>
Date: Thu, 26 Sep 2024 21:42:44 -0400
Subject: [PATCH] import sklearn.model_selection

---
 bnpm/automatic_regression.py | 1 +
 bnpm/cross_validation.py     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/bnpm/automatic_regression.py b/bnpm/automatic_regression.py
index bcb52d8..0db5c54 100644
--- 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
index 1b0d45a..a7c1ce3 100644
--- 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,