fix pytype error

google · Aug 22, 2023 · 7205cc1 · 7205cc1
1 parent 657f25b
commit 7205cc1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flax/linen/module.py b/flax/linen/module.py
@@ -891,7 +891,7 @@ def module_field(*, kw_only: bool = False, default: Optional[Any] = ...) -> Any:
 # * Other attributes are annotated for completeness. Because we are using
 #   the `if typing.TYPE_CHECKING` pattern, these annotations are not present
 #   at runtime so they don't affect the dataclass behavior.
-@dataclass_transform(field_specifiers=(module_field,))
+@dataclass_transform(field_specifiers=(module_field,))  # type: ignore[literal-required]
 class ModuleBase:
   if typing.TYPE_CHECKING:
     scope: Optional[Scope]