odify genome for the official release

tensorneat/test/crossover_mutation.py

@@ -1,5 +1,5 @@
 import jax, jax.numpy as jnp
-from utils import Act
+from tensorneat.common import Act
 from algorithm.neat import *
 import numpy as np
 

tensorneat/test/nan_fitness.py

@@ -1,5 +1,5 @@
 import jax, jax.numpy as jnp
-from utils import Act
+from tensorneat.common import Act
 from algorithm.neat import *
 import numpy as np
 

tensorneat/test/test_kan.ipynb

@@ -27,7 +27,7 @@
     "from algorithm.neat.gene.node.kan_node import KANNode\n",
     "from algorithm.neat.gene.conn.bspline import BSplineConn\n",
     "from problem.func_fit import XOR3d\n",
-    "from utils import Act\n",
+    "from tensorneat.utils import Act\n",
     "\n",
     "import jax, jax.numpy as jnp\n",
     "\n",

tensorneat/test/test_nan_fitness.py

@@ -1,5 +1,5 @@
 import jax, jax.numpy as jnp
-from utils import Act
+from tensorneat.common import Act
 from algorithm.neat import *
 import numpy as np
 

tensorneat/test/test_record_episode.ipynb

@@ -14,7 +14,7 @@
    "outputs": [],
    "source": [
     "import jax, jax.numpy as jnp\n",
-    "from utils import State\n",
+    "from tensorneat.utils import State\n",
     "from problem.rl_env import BraxEnv\n",
     "\n",
     "\n",

tensorneat/test/test_update_by_batch.ipynb

@@ -145,7 +145,7 @@
    "source": [
     "from algorithm.neat.gene.node.normalized import NormalizedNode\n",
     "from algorithm.neat.gene.conn import DefaultConnGene\n",
-    "from utils import Act\n",
+    "from tensorneat.utils import Act\n",
     "\n",
     "genome = DefaultGenome(num_inputs=3, num_outputs=2, max_nodes=10, max_conns=10,\n",
     "                       node_gene=NormalizedNode(activation_default=Act.identity, activation_options=(Act.identity,)),\n",