AttributeError when running EvoBind in colab notebook

[PepKindaChemist]

Hello Patrick, hope this message finds you well. When trying to use EvoBind on colab notebook, I keep running into an attribute error as follows:

AttributeError Traceback (most recent call last)
in <cell line: 27>()
25
26
---> 27 from mc_design_colab import main
28 MAX_RECYCLES=8 #max_recycles (default=3)
29 MODEL_NAME='model_1' #model_1_ptm

8 frames
/usr/local/lib/python3.10/dist-packages/jax/_src/deprecations.py in getattr(name)
50 message, fn = deprecations[name]
51 if fn is None: # Is the deprecation accelerated?
---> 52 raise AttributeError(message)
53 warnings.warn(message, DeprecationWarning, stacklevel=2)
54 return fn

AttributeError: jax.interpreters.xla.xe was removed in JAX v0.4.36. Use jax.lib.xla_extension instead.

I'm not sure if I need to update anything with the dependencies, or if an update broke the code.
Thanks!

[salvatta2]

Hi, I get also the same error. Has anyone found a solution to this?
Thanks

[patrickbryant1]

Hi, this is probably an issue due to jax updates. We will look into providing a fix in the coming weeks.

[salvatta2]

Thanks, appreciated!

…

On Fri, 27 Dec 2024 at 23:57, Patrick Bryant ***@***.***> wrote: Hi, this is probably an issue due to jax updates. We will look into providing a fix in the coming weeks. — Reply to this email directly, view it on GitHub <#29 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/A3MJICU7CJSG6E3CLWBZC232HXLO5AVCNFSM6AAAAABTJ4KWBKVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDKNRUGA3TQOBZG4> . You are receiving this because you commented.Message ID: ***@***.***>

[PepKindaChemist]

Thanks for the update @patrickbryant1, much appreciated!