fix bugs

tensorneat/algorithm/neat/__init__.py

@@ -1,4 +1,2 @@
-from .gene import *
-from .genome import *
 from .species import *
 from .neat import NEAT

tensorneat/algorithm/neat/species/base.py

@@ -1,5 +1,5 @@
 from tensorneat.common import State, StatefulBaseClass
-from ..genome import BaseGenome
+from tensorneat.genome import BaseGenome
 
 
 class BaseSpecies(StatefulBaseClass):

tensorneat/algorithm/neat/species/default.py

@@ -1,16 +1,17 @@
 import jax, jax.numpy as jnp
+
+from .base import BaseSpecies
 from tensorneat.common import (
     State,
     rank_elements,
     argmin_with_mask,
     fetch_first,
 )
-from ..genome.utils import (
+from tensorneat.genome.utils import (
     extract_conn_attrs,
     extract_node_attrs,
 )
-from ..genome import BaseGenome
-from .base import BaseSpecies
+from tensorneat.genome import BaseGenome
 
 
 """