Hi, Thank you for sharing this work. I am a biologist, and when I tried to run this on colab (with colab GPU) I always see Jax file issue. kindly help me. the full error is below:
/usr/local/lib/python3.12/dist-packages/sklearn/base.py:380: InconsistentVersionWarning: Trying to unpickle estimator SVC from version 1.2.2 when using version 1.6.1. This might lead to breaking code or invalid results. Use at your own risk. For more info please refer to:
https://scikit-learn.org/stable/model_persistence.html#security-maintainability-limitations
warnings.warn(
restraints:
no 1v1 restraints provided.
no 1vN restraints provided.
no MvN restraints provided.
no repulsive restraints provided.
Optimization losses include:
distogram loss, pLDDT, and ipAE.
Start optimization
AttributeError Traceback (most recent call last)
/tmp/ipykernel_484/1672137886.py in <cell line: 0>()
157 print(print_str)
158 print('\nStart optimization')
--> 159 dock_model.dock_rank()
30 frames
[... skipping hidden 23 frame]
[... skipping hidden 13 frame]
[... skipping hidden 13 frame]
/usr/local/lib/python3.12/dist-packages/jax/_src/deprecations.py in getattr(name)
52 message, fn = deprecations[name]
53 if fn is None: # Is the deprecation accelerated?
---> 54 raise AttributeError(message)
55 warnings.warn(message, DeprecationWarning, stacklevel=2)
56 return fn
AttributeError: jax.tree_unflatten was removed in JAX v0.6.0: use jax.tree.unflatten (jax v0.4.25 or newer) or jax.tree_util.tree_unflatten (any JAX version).
Hi, Thank you for sharing this work. I am a biologist, and when I tried to run this on colab (with colab GPU) I always see Jax file issue. kindly help me. the full error is below:
/usr/local/lib/python3.12/dist-packages/sklearn/base.py:380: InconsistentVersionWarning: Trying to unpickle estimator SVC from version 1.2.2 when using version 1.6.1. This might lead to breaking code or invalid results. Use at your own risk. For more info please refer to:
https://scikit-learn.org/stable/model_persistence.html#security-maintainability-limitations
warnings.warn(
restraints:
no 1v1 restraints provided.
no 1vN restraints provided.
no MvN restraints provided.
no repulsive restraints provided.
Optimization losses include:
distogram loss, pLDDT, and ipAE.
Start optimization
AttributeError Traceback (most recent call last)
/tmp/ipykernel_484/1672137886.py in <cell line: 0>()
157 print(print_str)
158 print('\nStart optimization')
--> 159 dock_model.dock_rank()
30 frames
[... skipping hidden 23 frame]
/usr/local/lib/python3.12/dist-packages/jax/_src/deprecations.py in getattr(name)
52 message, fn = deprecations[name]
53 if fn is None: # Is the deprecation accelerated?
---> 54 raise AttributeError(message)
55 warnings.warn(message, DeprecationWarning, stacklevel=2)
56 return fn
AttributeError: jax.tree_unflatten was removed in JAX v0.6.0: use jax.tree.unflatten (jax v0.4.25 or newer) or jax.tree_util.tree_unflatten (any JAX version).