complete fully stateful!

use black to format all files!

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

@@ -109,10 +109,10 @@ class DefaultNodeGene(BaseNodeGene):
 
     def distance(self, state, node1, node2):
         return (
-            jnp.abs(node1[1] - node2[1])
-            + jnp.abs(node1[2] - node2[2])
-            + (node1[3] != node2[3])
-            + (node1[4] != node2[4])
+            jnp.abs(node1[1] - node2[1])  # bias
+            + jnp.abs(node1[2] - node2[2])  # response
+            + (node1[3] != node2[3])  # activation
+            + (node1[4] != node2[4])  # aggregation
         )
 
     def forward(self, state, attrs, inputs, is_output_node=False):