Update dataclass_test to make CI pass.

PiperOrigin-RevId: 572535751
google-deepmind · Oct 11, 2023 · 7d5b0d0 · 7d5b0d0
1 parent 8929f6a
commit 7d5b0d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chex/_src/dataclass_test.py b/chex/_src/dataclass_test.py
@@ -119,10 +119,10 @@ class NestedClass:
     k_arr: np.ndarray
     k_dclass_with_map: Class
     k_dclass_no_map: ClassWithoutMap
-    k_dict_factory: dict = dataclasses.field(  # pylint:disable=g-bare-generic
+    k_dict_factory: dict = dataclasses.field(  # pylint:disable=g-bare-generic,invalid-field-call
         default_factory=lambda: dict(x='x', y='y'))
     k_default: str = 'default_str'
-    k_non_init: int = dataclasses.field(default=1, init=False)
+    k_non_init: int = dataclasses.field(default=1, init=False)  # pylint:disable=g-bare-generic,invalid-field-call
     k_init_only: dataclasses.InitVar[int] = 10
 
     def some_method(self, *args):