[dp] Decorate differential privacy configs w/ dataclass.

PiperOrigin-RevId: 437870878

official/modeling/privacy/configs.py

@@ -13,10 +13,12 @@
 # limitations under the License.
 
 """Configs for differential privacy."""
+import dataclasses
 
 from official.modeling.hyperparams import base_config
 
 
+@dataclasses.dataclass
 class DifferentialPrivacyConfig(base_config.Config):
   # Applied to the gradients
   # Setting to a large number so nothing is clipped.

official/modeling/privacy/configs_test.py

@@ -29,10 +29,10 @@ class ConfigsTest(tf.test.TestCase):
     self.assertEqual(0.0, noise_multiplier)
 
   def test_config(self):
-    dp_config = configs.DifferentialPrivacyConfig({
-        'clipping_norm': 1.0,
-        'noise_multiplier': 1.0
-    })
+    dp_config = configs.DifferentialPrivacyConfig(
+        clipping_norm=1.0,
+        noise_multiplier=1.0,
+    )
     self.assertEqual(1.0, dp_config.clipping_norm)
     self.assertEqual(1.0, dp_config.noise_multiplier)