search

pymc-devs / pytensor

PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
https://pytensor.readthedocs.io
Other
300 stars 91 forks source link

Update APIs to be compatible with NumPy 2.0 or pin NumPy to latest 1.x version #827

Open HangenYuu opened 3 weeks ago

HangenYuu commented 3 weeks ago

Description

NumPy 2.0 introduced some breaking changes in APIs, which break code and make tests unable to run. One example is np.obj2sctype(dtype) used in pytensor/tensor/type.py:105, which was removed in NumPy 2.0, thus interfering with test run for tests/link/jax/test_nlinalg.py. I think we need to update the APIs to be compatiable with NumPy 2.0, or pint NumPy to the latest 1.x version for now, NumPy is currently 2.0.0 when creating the environment from environment.yml.

ricardoV94 commented 3 weeks ago

Regarding the API, it's tracked by #688

The dev environment should pin it, I thought we were already doing it from #692 but maybe not the dev envs?