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/conn/base.py
+++ b/tensorneat/algorithm/neat/gene/conn/base.py
@@ -39,5 +39,5 @@ class BaseConnGene(BaseGene):
            "out": int(out_idx),
        }

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