diff --git a/README.md b/README.md index 2a0f4a8..b499b70 100644 --- a/README.md +++ b/README.md @@ -84,30 +84,30 @@ from tensorneat.common import ACT, AGG # define the pipeline pipeline = Pipeline( - algorithm=NEAT( - pop_size=1000, - species_size=20, - survival_threshold=0.1, - compatibility_threshold=1.0, - genome=DefaultGenome( - num_inputs=11, - num_outputs=3, - init_hidden_layers=(), - node_gene=BiasNode( - activation_options=ACT.tanh, - aggregation_options=AGG.sum, - ), - output_transform=ACT.tanh, + algorithm=NEAT( + pop_size=1000, + species_size=20, + survival_threshold=0.1, + compatibility_threshold=1.0, + genome=DefaultGenome( + num_inputs=11, + num_outputs=3, + init_hidden_layers=(), + node_gene=BiasNode( + activation_options=ACT.tanh, + aggregation_options=AGG.sum, ), + output_transform=ACT.tanh, ), - problem=BraxEnv( - env_name="hopper", - max_step=1000, - ), - seed=42, - generation_limit=100, - fitness_target=5000, - ) + ), + problem=BraxEnv( + env_name="hopper", + max_step=1000, + ), + seed=42, + generation_limit=100, + fitness_target=5000, +) # initialize state state = pipeline.setup()