ModuleNotFoundError for numpy >= 2.0.0

[scott2b]

Describe the issue as clearly as possible:

function_base appears now to be private and no longer directly accessible as done here: https://github.com/outlines-dev/outlines/blob/main/outlines/base.py#L8

As of numpy 2.0.0, this will raise:
ModuleNotFoundError: No module named 'numpy.lib.function_base'

Steps/code to reproduce the bug:

$ pip install 'numpy>=2.0.0
$ python

>>> import outlines

### Expected result:

```shell
the module imports

Error message:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/__init__.py", line 2, in <module>
    import outlines.generate
  File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/generate/__init__.py", line 2, in <module>
    from .cfg import cfg
  File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/generate/cfg.py", line 5, in <module>
    from outlines.models import OpenAI
  File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/models/__init__.py", line 14, in <module>
    from .openai import OpenAI, azure_openai, openai
  File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/models/openai.py", line 9, in <module>
    from outlines.base import vectorize
  File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/base.py", line 8, in <module>
    from numpy.lib.function_base import (
ModuleNotFoundError: No module named 'numpy.lib.function_base'

Outlines/Python version information:

Traceback (most recent call last):
File "", line 1, in
File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/init.py", line 2, in
import outlines.generate
File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/generate/init.py", line 2, in
from .cfg import cfg
File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/generate/cfg.py", line 5, in
from outlines.models import OpenAI
File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/models/init.py", line 14, in
from .openai import OpenAI, azure_openai, openai
File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/models/openai.py", line 9, in
from outlines.base import vectorize
File "/Users/scott/github/scott2b/work/outlines/.venv/lib/python3.12/site-packages/outlines/base.py", line 8, in
from numpy.lib.function_base import (
ModuleNotFoundError: No module named 'numpy.lib.function_base'
Python 3.12.3 (main, May 9 2024, 13:22:55) [Clang 15.0.0 (clang-1500.3.9.4)]
aiohttp==3.9.5
aiosignal==1.3.1
annotated-types==0.7.0
attrs==23.2.0
certifi==2024.6.2
charset-normalizer==3.3.2
cloudpickle==3.0.0
datasets==2.20.0
dill==0.3.8
diskcache==5.6.3
filelock==3.15.1
frozenlist==1.4.1
fsspec==2024.5.0
huggingface-hub==0.23.4
idna==3.7
interegular==0.3.3
Jinja2==3.1.4
jsonschema==4.22.0
jsonschema-specifications==2023.12.1
lark==1.1.9
llvmlite==0.43.0
MarkupSafe==2.1.5
multidict==6.0.5
multiprocess==0.70.16
nest-asyncio==1.6.0
numba==0.60.0
numpy==2.0.0
outlines @ git+https://github.com/outlines-dev/outlines.git@49146d5d30c3afc573a79fe37b02e9b58977285b
packaging==24.1
pandas==2.2.2
pyairports==2.1.1
pyarrow==16.1.0
pyarrow-hotfix==0.6
pycountry==24.6.1
pydantic==2.7.4
pydantic_core==2.18.4
python-dateutil==2.9.0.post0
pytz==2024.1
PyYAML==6.0.1
referencing==0.35.1
requests==2.32.3
rpds-py==0.18.1
setuptools==70.0.0
six==1.16.0
tqdm==4.66.4
typing_extensions==4.12.2
tzdata==2024.1
urllib3==2.2.1
xxhash==3.4.1
yarl==1.9.4

Context for the issue:

No response

[scott2b]

The workaround for the moment appears to be to install numpy<2.0.0

[rlouf]

Thank you for reporting this! The best solution for now would be to pin the version. A PR would be greatly appreciated!