Work around stubtest sklearn failure by temporariliy pinning

Avasam · Avasam · commit 302cb6122294 · 2025-06-20T14:44:31.000-04:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ tests = [
     # Needed for stubtest and downloads their dependencies to get known import symbols
     "networkx",
     "scikit-image",
-    "scikit-learn",
+    "scikit-learn <1.7.0", # TODO: Update stubs for sklearn
     "sympy",
     "vispy",
 ]
diff --git a/stubs/sklearn/stubtest_allowlist.txt b/stubs/sklearn/stubtest_allowlist.txt
@@ -1506,7 +1506,6 @@ sklearn.calibration.fmin_bfgs
 sklearn.calibration.is_classifier
 sklearn.calibration.partial
 sklearn.calibration.xlogy
-sklearn.cluster.AgglomerativeClustering.__init__
 sklearn.cluster.Birch.__init__
 sklearn.cluster.FeatureAgglomeration.fit_predict
 sklearn.cluster.KMeans.__init__

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ tests = [`
`26`	`26`	`# Needed for stubtest and downloads their dependencies to get known import symbols`
`27`	`27`	`"networkx",`
`28`	`28`	`"scikit-image",`
`29`		`- "scikit-learn",`
	`29`	`+ "scikit-learn <1.7.0", # TODO: Update stubs for sklearn`
`30`	`30`	`"sympy",`
`31`	`31`	`"vispy",`
`32`	`32`	`]`