From 9170b8824f69b08dd94d1dfd26e5c9910f8a93d8 Mon Sep 17 00:00:00 2001 From: keisuke-umezawa Date: Sun, 18 Feb 2024 15:11:24 +0900 Subject: [PATCH] Fix unittest --- optuna_dashboard/_serializer.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/optuna_dashboard/_serializer.py b/optuna_dashboard/_serializer.py index bbb94d97..ee527ca6 100644 --- a/optuna_dashboard/_serializer.py +++ b/optuna_dashboard/_serializer.py @@ -122,9 +122,7 @@ def serialize_frozen_study(study: FrozenStudy) -> dict[str, Any]: "study_name": study.study_name, "directions": [d.name.lower() for d in study.directions], "user_attrs": serialize_attrs(study.user_attrs), - "is_preferential": getattr(study, "_system_attrs", {}).get( - _SYSTEM_ATTR_PREFERENTIAL_STUDY, False - ), + "is_preferential": study.system_attrs.get(_SYSTEM_ATTR_PREFERENTIAL_STUDY, False), } return serialized @@ -146,7 +144,7 @@ def serialize_study_detail( "directions": [d.name.lower() for d in study.directions], "user_attrs": serialize_attrs(study.user_attrs), } - system_attrs = getattr(study, "system_attrs", {}) + system_attrs = study.system_attrs serialized["artifacts"] = list_study_artifacts(system_attrs) serialized["trials"] = [