AttributeError: module 'jax.interpreters.xla' has no attribute 'DeviceArray'

[amrzv]

Hi.
The dependencies should be changed (particularly the jax version shouldn't be the latest one 0.4.13).
Related issue in another repo.

Tested in colab

[bwarre471]

I can tell, this is fixed in Git repository, but this error only affects package installed trough pip.

Creators should upgrade pip package.

[bwarre471]

Quick cheap fix:

Edit the file dopamine/jax/networks.py @ line 351 change jax.interpreters.xla.DeviceArray to jax.Array and it'll run, or rebuild from github repo.

[psc-g]

apologies for the delay in response, we'll push a new version to pypi this week!

[matharg7]

@psc-g was this version pushed? this issue still exists

[psc-g]

apologies, will get this fixed today.

…

On Mon, Apr 29, 2024 at 8:37 PM Muhammad Athar ***@***.***> wrote: @psc-g <https://github.com/psc-g> was this version pushed? this issue still exists — Reply to this email directly, view it on GitHub <#211 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AE3CCMOXOQLRHGWU7YKTBE3Y73RV7AVCNFSM6AAAAAA24L5QBWVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDAOBTHE3DAMZUGM> . You are receiving this because you were mentioned.Message ID: <google/dopamine/issues/211/2083960343 ***@***.***>

[psc-g]

the issue is when you install from pip. i need to push a new version to pypi which should fix the issue, but it's blocked on some internal changes. should hopefully be able to get the fix out before the end of this week. apologies once again for the delay.

…

On Tue, Apr 30, 2024 at 5:23 AM Pablo Samuel Castro ***@***.***> wrote: apologies, will get this fixed today. On Mon, Apr 29, 2024 at 8:37 PM Muhammad Athar ***@***.***> wrote: > @psc-g <https://github.com/psc-g> was this version pushed? this issue > still exists > > — > Reply to this email directly, view it on GitHub > <#211 (comment)>, > or unsubscribe > <https://github.com/notifications/unsubscribe-auth/AE3CCMOXOQLRHGWU7YKTBE3Y73RV7AVCNFSM6AAAAAA24L5QBWVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDAOBTHE3DAMZUGM> > . > You are receiving this because you were mentioned.Message ID: > <google/dopamine/issues/211/2083960343 ***@***.***> >