From c17d3484d32293546b760a9be029d0d2d9853ec1 Mon Sep 17 00:00:00 2001
From: "Thomas S. Binns" <t.s.binns@outlook.com>
Date: Mon, 30 Dec 2024 11:56:52 +0000
Subject: [PATCH] [MAINT] Add scikit-learn as a dependency (#42)

---
 environment.yml |  1 +
 pyproject.toml  | 10 +++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/environment.yml b/environment.yml
index ed53b55..af4665d 100644
--- a/environment.yml
+++ b/environment.yml
@@ -8,4 +8,5 @@ dependencies:
   - mne>1.6
   - numba
   - numpy
+  - scikit-learn
   - scipy
diff --git a/pyproject.toml b/pyproject.toml
index 3afcae0..9995848 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,7 +9,15 @@ classifiers = [
   "Operating System :: OS Independent",
   "Programming Language :: Python :: 3",
 ]
-dependencies = ["joblib", "matplotlib", "mne>1.6", "numba", "numpy", "scipy"]
+dependencies = [
+  "joblib",
+  "matplotlib",
+  "mne>1.6",
+  "numba",
+  "numpy",
+  "scikit-learn",
+  "scipy",
+]
 description = "A Python signal processing package for computing spectral-domain and time-domain interactions using the bispectrum."
 name = "pybispectra"
 readme = "README.md"