Address sklearn device issues (#3748)

Saaketh Narayan · web-flow · commit 8f3a7ffefec8 · 2025-01-06T13:02:54.000-05:00
diff --git a/setup.py b/setup.py
@@ -143,6 +143,7 @@ def package_files(prefix: str, directory: str, extension: str):
     'cryptography==44.0.0',
     'pytest-httpserver>=1.0.4,<1.1',
     'setuptools<=59.5.0',
+    'scikit-learn>=1.2.0,<1.6',
 ]
 
 extra_deps['system_metrics_monitor'] = {

Original file line number	Diff line number	Diff line change
`@@ -143,6 +143,7 @@ def package_files(prefix: str, directory: str, extension: str):`
`143`	`143`	`'cryptography==44.0.0',`
`144`	`144`	`'pytest-httpserver>=1.0.4,<1.1',`
`145`	`145`	`'setuptools<=59.5.0',`
	`146`	`+ 'scikit-learn>=1.2.0,<1.6',`
`146`	`147`	`]`
`147`	`148`
`148`	`149`	`extra_deps['system_metrics_monitor'] = {`