mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-04-04 11:10:53 +00:00
fix deprecated FftType in jaxpm.kernels
This commit is contained in:
parent
5d34d3c3a8
commit
adaf7d236d
1 changed files with 1 additions and 1 deletions
|
@ -1,6 +1,6 @@
|
|||
import jax.numpy as jnp
|
||||
import numpy as np
|
||||
from jax.lib.xla_client import FftType
|
||||
from jax.lax import FftType
|
||||
from jax.sharding import PartitionSpec as P
|
||||
from jaxdecomp import fftfreq3d, get_output_specs
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue