odify genome for the official release

tensorneat/algorithm/neat/gene/base.py

@@ -1,5 +1,5 @@
 import jax, jax.numpy as jnp
-from utils import State, StatefulBaseClass, hash_array
+from tensorneat.common import State, StatefulBaseClass, hash_array
 
 
 class BaseGene(StatefulBaseClass):

tensorneat/algorithm/neat/gene/conn/default.py

@@ -2,7 +2,7 @@ import jax.numpy as jnp
 import jax.random
 import numpy as np
 import sympy as sp
-from utils import mutate_float
+from tensorneat.common import mutate_float
 from . import BaseConnGene
 
 

tensorneat/algorithm/neat/gene/node/default.py

@@ -4,7 +4,7 @@ import numpy as np
 import jax, jax.numpy as jnp
 import sympy as sp
 
-from utils import (
+from tensorneat.common import (
     Act,
     Agg,
     act_func,

tensorneat/algorithm/neat/gene/node/default_without_response.py

@@ -3,7 +3,7 @@ from typing import Tuple
 import jax, jax.numpy as jnp
 import numpy as np
 import sympy as sp
-from utils import (
+from tensorneat.common import (
     Act,
     Agg,
     act_func,

tensorneat/algorithm/neat/gene/node/kan_node.py

@@ -1,6 +1,6 @@
 import jax.numpy as jnp
 from . import BaseNodeGene
-from utils import Agg
+from tensorneat.common import Agg
 
 
 class KANNode(BaseNodeGene):

tensorneat/algorithm/neat/gene/node/min_max_node.py

@@ -2,7 +2,7 @@ from typing import Tuple
 
 import jax, jax.numpy as jnp
 
-from utils import Act, Agg, act_func, agg_func, mutate_int, mutate_float
+from tensorneat.common import Act, Agg, act_func, agg_func, mutate_int, mutate_float
 from . import BaseNodeGene
 
 

tensorneat/algorithm/neat/gene/node/normalized.py

@@ -2,7 +2,7 @@ from typing import Tuple
 
 import jax, jax.numpy as jnp
 
-from utils import Act, Agg, act_func, agg_func, mutate_int, mutate_float
+from tensorneat.common import Act, Agg, act_func, agg_func, mutate_int, mutate_float
 from . import BaseNodeGene