search

google / dopamine

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
https://github.com/google/dopamine
Apache License 2.0
10.42k stars 1.36k forks source link

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

Open amrzv opened 11 months ago

amrzv commented 11 months ago

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

bwarre471 commented 10 months ago

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

Creators should upgrade pip package.

bwarre471 commented 10 months ago

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 commented 10 months ago

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

matharg7 commented 2 months ago

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

psc-g commented 1 month ago

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 https://github.com/google/dopamine/issues/211#issuecomment-2083960343, 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 commented 1 month ago

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 https://github.com/google/dopamine/issues/211#issuecomment-2083960343, or unsubscribe https://github.com/notifications/unsubscribe-auth/AE3CCMOXOQLRHGWU7YKTBE3Y73RV7AVCNFSM6AAAAAA24L5QBWVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDAOBTHE3DAMZUGM . You are receiving this because you were mentioned.Message ID: <google/dopamine/issues/211/2083960343 @.***>