refactor names;

delete useless

examples/a.py

@@ -1,33 +0,0 @@
-import jax.random
-import numpy as np
-import jax.numpy as jnp
-import time
-
-
-def random_array(key):
-    return jax.random.normal(key, (1000,))
-
-def random_array_np():
-    return np.random.normal(size=(1000,))
-
-
-def t_jax():
-    key = jax.random.PRNGKey(42)
-    max_li = []
-    tic = time.time()
-    for _ in range(100):
-        key, sub_key = jax.random.split(key)
-        array = random_array(sub_key)
-        array = jax.device_get(array)
-        max_li.append(max(array))
-    print(max_li, time.time() - tic)
-
-def t_np():
-    max_li = []
-    tic = time.time()
-    for _ in range(100):
-        max_li.append(max(random_array_np()))
-    print(max_li, time.time() - tic)
-
-if __name__ == '__main__':
-    t_np()

examples/func_fit/xor.py

@@ -1,5 +1,5 @@
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from problem.func_fit import XOR, FuncFitConfig

examples/func_fit/xor_hyperneat.py

@@ -1,5 +1,5 @@
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm.neat import NormalGene, NormalGeneConfig
 from algorithm.hyperneat import HyperNEAT, NormalSubstrate, NormalSubstrateConfig
 from problem.func_fit import XOR3d, FuncFitConfig

examples/func_fit/xor_recurrent.py

@@ -1,5 +1,5 @@
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import RecurrentGene, RecurrentGeneConfig
 from problem.func_fit import XOR3d, FuncFitConfig

examples/gymnax/acrobot.py

@@ -1,7 +1,7 @@
 import jax.numpy as jnp
 
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from problem.rl_env import GymNaxConfig, GymNaxEnv

examples/gymnax/cartpole.py

@@ -1,7 +1,7 @@
 import jax.numpy as jnp
 
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from problem.rl_env import GymNaxConfig, GymNaxEnv

examples/gymnax/cartpole_hyperneat.py

@@ -1,7 +1,7 @@
 import jax.numpy as jnp
 
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from algorithm.hyperneat import HyperNEAT, NormalSubstrateConfig, NormalSubstrate

examples/gymnax/mountain_car.py

@@ -1,7 +1,7 @@
 import jax.numpy as jnp
 
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from problem.rl_env import GymNaxConfig, GymNaxEnv

examples/gymnax/mountain_car_continuous.py

@@ -1,7 +1,7 @@
 import jax.numpy as jnp
 
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from problem.rl_env import GymNaxConfig, GymNaxEnv

examples/gymnax/pendulum.py

@@ -1,7 +1,7 @@
 import jax.numpy as jnp
 
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from problem.rl_env import GymNaxConfig, GymNaxEnv

examples/gymnax/reacher.py

@@ -1,5 +1,5 @@
 from config import *
-from pipeline_jitable_env import Pipeline
+from pipeline import Pipeline
 from algorithm import NEAT
 from algorithm.neat.gene import NormalGene, NormalGeneConfig
 from problem.rl_env import GymNaxConfig, GymNaxEnv