From edfbd8aad20a79091e429fb6093fb9fa2518217a Mon Sep 17 00:00:00 2001 From: Ashutosh0x Date: Tue, 6 Jan 2026 20:26:13 +0530 Subject: [PATCH] fix(dataclass): enable kw_only=True by default to fix inheritance (closes #305) --- chex/_src/dataclass.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/chex/_src/dataclass.py b/chex/_src/dataclass.py index ea111259..e2fdf29f 100644 --- a/chex/_src/dataclass.py +++ b/chex/_src/dataclass.py @@ -100,7 +100,7 @@ def dataclass( order=False, unsafe_hash=False, frozen=False, - kw_only: bool = False, + kw_only: bool = True, mappable_dataclass=True, # pylint: disable=redefined-outer-name ): """JAX-friendly wrapper for :py:func:`dataclasses.dataclass`. @@ -154,7 +154,7 @@ def __init__( order=False, unsafe_hash=False, frozen=False, - kw_only=False, + kw_only=True, mappable_dataclass=True, # pylint: disable=redefined-outer-name ): self.init = init