Unable to run the examples

[oscarodeigah]

Hi Ivan,

First off, thanks for developing this package. I have been looking for a package that would allow me to combine the capabilities of JAX with Fenics and I believe you have just provided this solution.
I tried running the examples you provided, but I keep getting the error message below:

TypeError: init(): incompatible constructor arguments. The following argument types are supported:
1. dolfin.cpp.la.LUSolver()
2. dolfin.cpp.la.LUSolver(A: dolfin.cpp.la.GenericLinearOperator, method: str='default')
3. dolfin.cpp.la.LUSolver(comm: MPICommWrapper, A: dolfin.cpp.la.GenericLinearOperator, method: str='default')

Invoked with: 'default'

I also tried running the tests to be sure the program was installed completely or correctly, but I got the same error message:

FAILED tests/fenics/test_solve_rev.py::test_jacobian_and_vjp - TypeError: init(): incompatible constructor arguments. The following ar...
FAILED tests/fenics/test_solve_rev.py::test_grad - TypeError: init(): incompatible constructor arguments. The following argument types...

Have you experienced similar issues or is there something I am doing wrong?
I look forward to your reply.

Best regards,
Oscar