update problem and pipeline

2024-07-11 19:34:12 +08:00
parent be6a67d7e2
commit cef27b56bb
14 changed files with 40 additions and 205 deletions
--- a/tensorneat/problem/func_fit/custom.py
+++ b/tensorneat/problem/func_fit/custom.py
@@ -1,6 +1,4 @@
-from typing import Callable, Union, List, Tuple, Sequence
-
-import jax
+from typing import Callable, Union, List, Tuple
 from jax import vmap, Array, numpy as jnp
 import numpy as np