From a7a1bf9b5f90be12ec6ebcbbb2a788e7be1a4af0 Mon Sep 17 00:00:00 2001
From: Oleksandr Shchur <shchuro@amazon.com>
Date: Tue, 28 Apr 2026 08:43:01 +0000
Subject: [PATCH] Remove sklearn dependency

---
 pyproject.toml | 1 -
 1 file changed, 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index d9e7117..78e7eca 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,7 +19,6 @@ 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",