From 6bfdb4031dec751d51b6e68c09c768bd41922291 Mon Sep 17 00:00:00 2001 From: Byron Hsu Date: Sat, 5 Oct 2024 11:07:41 -0700 Subject: [PATCH] [Easy] use .text() instead of .text (#1577) --- python/sglang/test/test_programs.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python/sglang/test/test_programs.py b/python/sglang/test/test_programs.py index 41f466f73..092c5369d 100644 --- a/python/sglang/test/test_programs.py +++ b/python/sglang/test/test_programs.py @@ -72,7 +72,7 @@ def test_select(check_answer): statement="The capital of Germany is Berlin.", ) if check_answer: - assert ret["answer"] == "True", ret.text + assert ret["answer"] == "True", ret.text() else: assert ret["answer"] in ["True", "False", "Unknown"] @@ -80,7 +80,7 @@ def test_select(check_answer): statement="The capital of Canada is Tokyo.", ) if check_answer: - assert ret["answer"] == "False", ret.text + assert ret["answer"] == "False", ret.text() else: assert ret["answer"] in ["True", "False", "Unknown"] @@ -88,7 +88,7 @@ def test_select(check_answer): statement="Purple is a better color than green.", ) if check_answer: - assert ret["answer"] == "Unknown", ret.text + assert ret["answer"] == "Unknown", ret.text() else: assert ret["answer"] in ["True", "False", "Unknown"] @@ -100,8 +100,8 @@ def test_decode_int(): s += "The number of days in a year is " + sgl.gen_int("days") + "\n" ret = decode_int.run(temperature=0.1) - assert int(ret["hours"]) == 24, ret.text - assert int(ret["days"]) == 365, ret.text + assert int(ret["hours"]) == 24, ret.text() + assert int(ret["days"]) == 365, ret.text() def test_decode_json_regex():