diff --git a/fiddle/_src/experimental/auto_config.py b/fiddle/_src/experimental/auto_config.py
index 029cc006..b91f224e 100644
--- a/fiddle/_src/experimental/auto_config.py
+++ b/fiddle/_src/experimental/auto_config.py
@@ -440,7 +440,7 @@ def fn(...):  # Or some expression involving a lambda.
 
   wrapper_module = ast.Module(
       body=[
-          ast.FunctionDef(
+          ast.FunctionDef(  # pytype: disable=missing-parameter
               name=_CLOSURE_WRAPPER_ID,
               args=_EMPTY_ARGUMENTS,
               body=[