complete fully stateful!

use black to format all files!

tensorneat/examples/brax/walker.py

@@ -4,7 +4,7 @@ from algorithm.neat import *
 from problem.rl_env import BraxEnv
 from utils import Act
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     pipeline = Pipeline(
         algorithm=NEAT(
             species=DefaultSpecies(
@@ -16,21 +16,21 @@ if __name__ == '__main__':
                     node_gene=DefaultNodeGene(
                         activation_options=(Act.tanh,),
                         activation_default=Act.tanh,
-                    )
+                    ),
                 ),
                 pop_size=10000,
                 species_size=10,
             ),
         ),
         problem=BraxEnv(
-            env_name='walker2d',
+            env_name="walker2d",
         ),
         generation_limit=10000,
-        fitness_target=5000
+        fitness_target=5000,
     )
 
     # initialize state
     state = pipeline.setup()
     # print(state)
     # run until terminate
-    state, best = pipeline.auto_run(state)
+    state, best = pipeline.auto_run(state)