update gene

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

@@ -12,34 +12,6 @@ class BaseNodeGene(BaseGene):
     def forward(self, state, attrs, inputs, is_output_node=False):
         raise NotImplementedError
 
-    def input_transform(self, state, attrs, inputs):
-        """
-        make transformation in the input node.
-        default: do nothing
-        """
-        return inputs
-
-    def update_by_batch(self, state, attrs, batch_inputs, is_output_node=False):
-        # default: do not update attrs, but to calculate batch_res
-        return (
-            jax.vmap(self.forward, in_axes=(None, None, 0, None))(
-                state, attrs, batch_inputs, is_output_node
-            ),
-            attrs,
-        )
-
-    def update_input_transform(self, state, attrs, batch_inputs):
-        """
-        update the attrs for transformation in the input node.
-        default: do nothing
-        """
-        return (
-            jax.vmap(self.input_transform, in_axes=(None, None, 0))(
-                state, attrs, batch_inputs
-            ),
-            attrs,
-        )
-
     def repr(self, state, node, precision=2, idx_width=3, func_width=8):
         idx = node[0]