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/forward.py
+++ b/algorithms/neat/genome/forward.py
@@ -1,6 +1,5 @@
 import jax
-from jax import Array, numpy as jnp
-from jax import jit, vmap
+from jax import Array, numpy as jnp, jit, vmap

 from .utils import I_INT