From c9cf37b11e9d93ee469247bd56203f4774e8e318 Mon Sep 17 00:00:00 2001 From: James Martens Date: Tue, 1 Jul 2025 04:03:55 -0700 Subject: [PATCH] Improved handling of dks_params and tat_params arguments. PiperOrigin-RevId: 777972116 --- dks/base/activation_transform.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/dks/base/activation_transform.py b/dks/base/activation_transform.py index b06f61d..92786ba 100644 --- a/dks/base/activation_transform.py +++ b/dks/base/activation_transform.py @@ -460,8 +460,15 @@ def _get_activations_params( # we do this instead: if dks_params is None: dks_params = {} + else: + # Make a copy and possibly convert from ConfigDict + dks_params = dict(dks_params) + if tat_params is None: tat_params = {} + else: + # Make a copy and possibly convert from ConfigDict + tat_params = dict(tat_params) _verify_params_dict_and_set_defaults( dks_params, {"c_slope_1_target": 1.5, "local_q_slope_target": 1.0})