Use name and value instead of the string representation of assignment

kubeflow · Aug 14, 2021 · b2085a6 · b2085a6
1 parent 1c0e15f
commit b2085a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/suggestion/v1beta1/optuna/service.py b/pkg/suggestion/v1beta1/optuna/service.py
@@ -141,7 +141,7 @@ def _tell(self, trials):
 
     def _get_assignments_key(self, assignments):
         assignments = sorted(assignments, key=lambda a: a.name)
-        assignments_str = [str(a) for a in assignments]
+        assignments_str = [f"{a.name}:{a.value}" for a in assignments]
         return ",".join(assignments_str)
 
     def _get_optuna_search_space(self):