From 9093f845fae40c8786c69abd224c191786bd5848 Mon Sep 17 00:00:00 2001 From: moririn2528 Date: Fri, 15 Sep 2023 17:51:05 +0900 Subject: [PATCH] fix by pytest error --- python_tests/test_api.py | 2 +- python_tests/test_preference_setting.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python_tests/test_api.py b/python_tests/test_api.py index db79c0d8..10687752 100644 --- a/python_tests/test_api.py +++ b/python_tests/test_api.py @@ -211,7 +211,7 @@ class APITestCase(TestCase): self.assertEqual(status, 200) study_detail = json.loads(body) - assert study_detail["feedback_component_type"]["type"] == "artifact" + assert study_detail["feedback_component_type"]["output_type"] == "artifact" assert study_detail["feedback_component_type"]["artifact_key"] == "image" def test_skip_trial(self) -> None: diff --git a/python_tests/test_preference_setting.py b/python_tests/test_preference_setting.py index a0adcb39..5499e134 100644 --- a/python_tests/test_preference_setting.py +++ b/python_tests/test_preference_setting.py @@ -14,7 +14,7 @@ class FeedbackSettingTestCase(TestCase): register_preference_feedback_component(study, "artifact", "image_key") system_attrs = study._study.system_attrs feedback_type = system_attrs.get(_SYSTEM_ATTR_FEEDBACK_COMPONENT, {}) - assert "type" in feedback_type - assert feedback_type["type"] == "artifact" + assert "output_type" in feedback_type + assert feedback_type["output_type"] == "artifact" assert "artifact_key" in feedback_type assert feedback_type["artifact_key"] == "image_key"