search

tumaer / JAXFLUIDS

Differentiable Fluid Dynamics Package
Other
298 stars 52 forks source link

import error #6

Closed ChengXiaoSi closed 2 months ago

ChengXiaoSi commented 5 months ago

Dear developers I get an error when I use from jaxfluids.simulation_manager import SimulationManager。The reson of this error is that from jax.config import config is used in /JAXFLUIDS/src/jaxfluids/simulation_manager.py.In fact, jax suggest using this way. import jax jax.config.update(your_config_options_here) Thank you! image

S-Yuan137 commented 5 months ago

Downgrading jax==0.4.24 from 0.4.25 works for me. No need to change the import header in simulation_manager.py

ChengXiaoSi commented 5 months ago

Downgrading jax==0.4.24 from 0.4.25 works for me. No need to change the import header in simulation_manager.py

Thanks for your advice