Skip to content

Commit 38a5771

Browse files
committed
fix self error
Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>
1 parent 1f2a669 commit 38a5771

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

test/sglang/sagemaker/test_sm_endpoint.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -143,10 +143,8 @@ def model_endpoint(aws_session, model_package, instance_type):
143143

144144
@pytest.mark.parametrize("instance_type", ["ml.g5.12xlarge"])
145145
@pytest.mark.parametrize("model_id", ["Qwen/Qwen3-0.6B"], indirect=True)
146-
def test_sglang_sagemaker_endpoint(
147-
self, model_endpoint, model_id, max_tokens=2400, temperature=0.01
148-
):
149-
predictor, endpoint_name = model_endpoint
146+
def test_sglang_sagemaker_endpoint(model_endpoint, model_id, max_tokens=2400, temperature=0.01):
147+
predictor = model_endpoint
150148
prompt = "Write a python script to calculate square of n"
151149

152150
payload = {

0 commit comments

Comments
 (0)