optimize import

This commit is contained in:
wls2002
2023-06-29 09:41:49 +08:00
parent d28cef1a87
commit 01b7731231
14 changed files with 29 additions and 58 deletions

View File

@@ -4,8 +4,7 @@ contains operations on the population: creating the next generation and populati
import jax
from jax import jit, vmap, Array, numpy as jnp
from .genome import distance, mutate, crossover
from .genome.utils import I_INT, fetch_first, rank_elements
from .genome import distance, mutate, crossover, I_INT, fetch_first, rank_elements
@jit