From 9d59eabd408ec06bbd399ec3cc9410af0458a49b Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Thu, 10 Feb 2022 22:32:36 -0800
Subject: [PATCH] Add direct `sklearn ` dependency.

PiperOrigin-RevId: 427925353
---
 requirements.txt | 1 +
 setup.py         | 1 +
 2 files changed, 2 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index 36aab79..31929da 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,6 +6,7 @@ mpmath
 numpy~=1.21.5
 pandas~=1.1.4
 scipy~=1.2.1
+sklearn~=1.0.2
 tensorflow-datasets>=4.4.0
 tensorflow-estimator>=2.3.0
 tensorflow-probability>=0.13.0
diff --git a/setup.py b/setup.py
index 49c4330..393b85f 100644
--- a/setup.py
+++ b/setup.py
@@ -34,6 +34,7 @@ setup(
         'numpy~=1.21.5',
         'pandas~=1.1.4',
         'scipy~=1.2.1',
+        'sklearn~=1.0.2',
         'tensorflow-datasets>=4.4.0',
         'tensorflow-estimator>=2.3.0',
         'tensorflow-probability>=0.13.0',