BUG: NonlinearVariationalSolver errors with adj_kwargs

Abierto Apto para principiantes
  • ¿Algo no está bien?
#5,186 0 comentarios 0 reacciones 0 asignados Ver en GitHub

Los mantenedores suelen responder en 1 día

@dham ya está trabajando en esto.

Desde el 17/6/2026.

  • #5187 de @dham — abierto

Evaluación

Dificultad
2/5
Tiempo estimado
1-3 horas
Aptitud para principiantes
75/100
Tipo de issue
Error
Claridad
Bien especificado
Estado de actividad
Tranquilo
Stack tecnológico
python
Área
api, backend

Línea de trabajo

Comienza con adjoint_utils/variational_solver.py alrededor del wrapper de la línea 47 y luego ejecuta el ejemplo mínimo de Firedrake proporcionado para reproducir el error del constructor. Confirma que NonlinearVariationalSolver acepta adj_kwargs y que el ejemplo funciona sin depender del workaround en tiempo de solve.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

bug

Describe the bug
Passing adj_kwargs to NonlinearVariationalSolver in order to have different solver_parameters for the adjoint problem errors.

Steps to Reproduce
Here is an MFE:

from firedrake import *
from firedrake.adjoint import *

continue_annotation()

mesh = UnitSquareMesh(20,20)
U = FunctionSpace(mesh, "CG", 1)

u = Function(U)
v = TestFunction(U)

f = Function(U)

F = inner(grad(u),grad(v))*dx - inner(f,v)*dx

bcs = [DirichletBC(U, Constant(1), (4,)),
       DirichletBC(U, Constant(0), (1, 2, 3))]

sp_lu = {
        "snes_monitor": None,
        "pc_type": "lu",
        "pc_factor_mat_solver_type": "mumps",}

# This works
solve(F==0, u, bcs=bcs, solver_parameters=sp_lu, adj_kwargs={"solver_parameters": sp_lu})

nvp = NonlinearVariationalProblem(F, u, bcs=bcs)

# This errors
nvs = NonlinearVariationalSolver(nvp, solver_parameters=sp_lu,
                                 adj_kwargs={"solver_parameters": sp_lu})

Expected behavior
I expect the NonlinearVariationalSolver to accept the adj_kwargs argument.

Error message

Traceback (most recent call last):
  File "/home/user/Documents/Code/codex/debug_adjoint_nvs_2.py", line 36, in <module>
    nvs = NonlinearVariationalSolver(nvp, solver_parameters=sp_lu,
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "petsc4py/PETSc/Log.pyx", line 250, in petsc4py.PETSc.Log.EventDecorator.decorator.wrapped_func
  File "petsc4py/PETSc/Log.pyx", line 251, in petsc4py.PETSc.Log.EventDecorator.decorator.wrapped_func
  File "/usr/lib/python3.12/contextlib.py", line 81, in inner
    return func(*args, **kwds)
           ^^^^^^^^^^^^^^^^^^^
  File "/home/user/firedrake-release-30-05-2026/venv-firedrake/lib/python3.12/site-packages/firedrake/adjoint_utils/variational_solver.py", line 47, in wrapper
    init(self, problem, *args, **kwargs)
TypeError: NonlinearVariationalSolver.__init__() got an unexpected keyword argument 'adj_kwargs'

Environment:

  • OS: Linux
  • Python version: 3.12.3

Additional Info
As pointed out by @dham, the workaround is currently to pass adj_kwargs to the first solve as follows:

nvs = NonlinearVariationalSolver(nvp, solver_parameters=sp_lu)
nvs.solve(adj_kwargs={"solver_parameters": sp_lu})
Lenguaje dominante
Python
Estrellas
684
Forks
204
Merge medio
4 d 14 h
PR fusionados (30 d)
46

Preparar el entorno

  • Sin Dockerfile ni archivo de Docker Compose
  • Tiene una plantilla de pull request
  • Sin guía de contribución

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de firedrakeproject/firedrake

Todos los issues de firedrakeproject/firedrake

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.