optimize import

2023-06-29 09:41:49 +08:00
parent d28cef1a87
commit 01b7731231
14 changed files with 29 additions and 58 deletions
--- a/algorithms/neat/genome/distance.py
+++ b/algorithms/neat/genome/distance.py
@@ -5,8 +5,7 @@ See https://github.com/CodeReclaimers/neat-python/blob/master/neat/genome.py
 """
 from typing import Dict

-from jax import jit, vmap, Array
-from jax import numpy as jnp
+from jax import jit, vmap, Array, numpy as jnp

 from .utils import EMPTY_NODE, EMPTY_CON