add backend="jax" to sympy module

2024-06-13 05:55:33 +08:00
parent b3e442c688
commit 69d73aab73
12 changed files with 254 additions and 167 deletions
--- a/tensorneat/algorithm/neat/gene/node/base.py
+++ b/tensorneat/algorithm/neat/gene/node/base.py
@@ -54,5 +54,5 @@ class BaseNodeGene(BaseGene):
            "idx": int(idx),
        }

-    def sympy_func(self, state, node_dict, inputs, is_output_node=False, precision=None):
+    def sympy_func(self, state, node_dict, inputs, is_output_node=False):
        raise NotImplementedError