from af2seq.design.utils import generate_start_sequence
Traceback (most recent call last):
File "", line 1, in
File "/hpfs/userws/zhanz241/af2seq/af2seq/af2seq/init.py", line 1, in
from .design.core import GradientDesign,MCMCDesign
File "/hpfs/userws/zhanz241/af2seq/af2seq/af2seq/design/core.py", line 6, in
import optax
ModuleNotFoundError: No module named 'optax'
I am having difficulty loading optax even though I can see optax in my env:
"
opt_einsum 3.3.0 pyhd3eb1b0_1
optax 0.1.0 pyhd8ed1ab_0 conda-forge
packaging 24.1 py38h06a4308_0
"
Any suggestion as how to resolve this? Thank you in advance.
I am having difficulty loading optax even though I can see optax in my env: " opt_einsum 3.3.0 pyhd3eb1b0_1
optax 0.1.0 pyhd8ed1ab_0 conda-forge packaging 24.1 py38h06a4308_0
"
Any suggestion as how to resolve this? Thank you in advance.