make fully stateful in module gene.

2024-05-25 16:13:41 +08:00
parent 3b2f917aee
commit 625c261a49
5 changed files with 21 additions and 36 deletions
--- a/tensorneat/algorithm/neat/gene/node/base.py
+++ b/tensorneat/algorithm/neat/gene/node/base.py
@@ -8,5 +8,5 @@ class BaseNodeGene(BaseGene):
    def __init__(self):
        super().__init__()

-    def forward(self, attrs, inputs, is_output_node=False):
+    def forward(self, state, attrs, inputs, is_output_node=False):
        raise NotImplementedError