small change for elegant code style

This commit is contained in:
wls2002
2023-07-19 16:38:43 +08:00
parent a684e6584d
commit 80ee5ea2ea
5 changed files with 7 additions and 87 deletions

View File

@@ -1,5 +1,3 @@
from typing import Tuple
import jax
from jax import jit, Array, numpy as jnp