AttributeError: module 'jax.numpy' has no attribute 'DeviceArray'

liesel-devs / liesel

A probabilistic programming framework

https://liesel-project.org

MIT License

38 stars 2 forks source link

AttributeError: module 'jax.numpy' has no attribute 'DeviceArray' #104

Closed Seb-Lorek closed 10 months ago

Seb-Lorek commented 10 months ago

Importing liesel.goose results in error due to depreciation of DeviceArray for jax >= 0.4.14. Depreciated: jax.numpy.DeviceArray: use {class}jax.Array, see here for details.

Seb-Lorek commented 10 months ago

Update liesel >= 0.2.4, see.