remove create_func....

2023-08-02 13:26:01 +08:00
parent 85318f98f3
commit 1499e062fe
34 changed files with 558 additions and 1022 deletions
--- a/algorithm/neat/species/species_info.py
+++ b/algorithm/neat/species/species_info.py
@@ -2,6 +2,7 @@ from jax.tree_util import register_pytree_node_class
 import numpy as np
 import jax.numpy as jnp

+
@register_pytree_node_class
 class SpeciesInfo:

@@ -44,7 +45,6 @@ class SpeciesInfo:
    def size(self):
        return self.species_keys.shape[0]

-
    def tree_flatten(self):
        children = self.species_keys, self.best_fitness, self.last_improved, self.member_count
        aux_data = None