fix(ingest): pin numpy<2 for classification (#10725)

This commit is contained in:
Harshal Sheth 2024-06-17 13:50:08 -07:00 committed by GitHub
parent 82204eb451
commit 0816e7590c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -107,6 +107,10 @@ classification_lib = {
# This is a bit of a hack. Because we download the SpaCy model at runtime in the classify plugin, # This is a bit of a hack. Because we download the SpaCy model at runtime in the classify plugin,
# we need pip to be available. # we need pip to be available.
"pip", "pip",
# We were seeing an error like this `numpy.dtype size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject`
# with numpy 2.0. This likely indicates a mismatch between scikit-learn and numpy versions.
# https://stackoverflow.com/questions/40845304/runtimewarning-numpy-dtype-size-changed-may-indicate-binary-incompatibility
"numpy<2",
} }
sql_common = ( sql_common = (