From 2f8ff434f3c8a8395c006891fb0aca40c496af1e Mon Sep 17 00:00:00 2001 From: hrntsm Date: Fri, 21 Jul 2023 17:01:58 +0900 Subject: [PATCH] Update constants key to work with v2.1 --- optuna_dashboard/_serializer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optuna_dashboard/_serializer.py b/optuna_dashboard/_serializer.py index 10bdc8e5..ef785cbf 100644 --- a/optuna_dashboard/_serializer.py +++ b/optuna_dashboard/_serializer.py @@ -8,7 +8,6 @@ from typing import Union import numpy as np from optuna.distributions import BaseDistribution from optuna.distributions import CategoricalDistribution -from optuna.samplers._base import _CONSTRAINTS_KEY from optuna.study import StudySummary from optuna.trial import FrozenTrial @@ -84,6 +83,7 @@ if TYPE_CHECKING: MAX_ATTR_LENGTH = 1024 +CONSTRAINTS_KEY = "constraints" def serialize_attrs(attrs: dict[str, Any]) -> list[Attribute]: @@ -188,7 +188,7 @@ def serialize_frozen_trial( ), "note": note.get_note_from_system_attrs(study_system_attrs, trial._trial_id), "artifacts": list_trial_artifacts(study_system_attrs, trial._trial_id), - "constraints": trial_system_attrs.get(_CONSTRAINTS_KEY, []), + "constraints": trial_system_attrs.get(CONSTRAINTS_KEY, []), } serialized_intermediate_values: list[IntermediateValue] = []