fix: fix AlgorithmEstimator init

aliyun · Jul 11, 2024 · 63b8191 · 63b8191
1 parent e27d1e7
commit 63b8191
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pai/estimator.py b/pai/estimator.py
@@ -1164,9 +1164,9 @@ def _get_hyperparameters(
         if hps_def:
             # Get default hyperparameters.
             for hp in hps_def:
-                hp_name = hp.get("Name")
-                hp_value = hp.get("DefaultValue", "")
-                hp_type = hp.get("Type", "String")
+                hp_name = hp.name
+                hp_value = hp.default_value
+                hp_type = hp.type or "String"
                 # For hyperparameters with type INT or FLOAT, if the default value is
                 # empty, skip it.
                 if (