From 29076435284fef691292971ba944af2866acbf6a Mon Sep 17 00:00:00 2001
From: Oleksandr Shchur <shchuro@amazon.com>
Date: Tue, 28 Apr 2026 08:52:34 +0000
Subject: [PATCH] Keep sklearn

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 78e7eca..d9e7117 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,6 +19,7 @@ dependencies = [
   "accelerate>=0.34,<2",
   "numpy>=1.21,<3",
   "einops>=0.7.0,<1",
+  "scikit-learn>=1.6.0,<2",
 ]
 classifiers = [
   "Programming Language :: Python :: 3",