diff --git a/awe_components/setup/data.py b/awe_components/setup/data.py index 3536dca..626a5ea 100644 --- a/awe_components/setup/data.py +++ b/awe_components/setup/data.py @@ -1,6 +1,9 @@ #!/usr/bin/env python3 # Copyright 2022, Educational Testing Service +# TODO: using numpy>1.26.4 seems to cause errors when installing from awe_workbench. +# This is related to the desired model (subwordbert) that we use for AWE. +# Version-locking seems to keep this from failing. from spacy.cli.download import download import os diff --git a/setup.cfg b/setup.cfg index e37b5b7..66901e6 100644 --- a/setup.cfg +++ b/setup.cfg @@ -31,13 +31,12 @@ python_requires = >=3.9 cmdclass = install = install.AWEInstall install_requires = - awe_lexica + awe_lexica @ git+https://github.com/ArgLab/AWE_Lexica.git spacy - holmes_extractor coreferee rdflib spacytextblob - numpy + numpy==1.26.4 srsly wordfreq statistics @@ -45,13 +44,13 @@ install_requires = scikit-learn nltk aenum - websockets + websockets<14 websocket-client pytest clint pygtrie transformers - torch + torch==2.4.1 Path [options.package_data] * = *.cfg, *.csv, *.json, *.txt