From a32aeb688a37df0035b053eb1adbc2208dcdb9c8 Mon Sep 17 00:00:00 2001 From: Michael <13900043+michaelzhang-ai@users.noreply.github.com> Date: Mon, 8 Jun 2026 21:53:14 -0700 Subject: [PATCH] [AMD] Fix AttributeError in GeneratedSharedPrefixDataset.from_args for in-process callers (#27580) --- python/sglang/benchmark/datasets/generated_shared_prefix.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/benchmark/datasets/generated_shared_prefix.py b/python/sglang/benchmark/datasets/generated_shared_prefix.py index c739b5533..6b5ac3426 100644 --- a/python/sglang/benchmark/datasets/generated_shared_prefix.py +++ b/python/sglang/benchmark/datasets/generated_shared_prefix.py @@ -55,8 +55,8 @@ class GeneratedSharedPrefixDataset(BaseDataset): @classmethod def from_args(cls, args: Namespace) -> "GeneratedSharedPrefixDataset": assert not getattr(args, "tokenize_prompt", False) - group_distribution = args.gsp_group_distribution - zipf_alpha = args.gsp_zipf_alpha + group_distribution = getattr(args, "gsp_group_distribution", "uniform") + zipf_alpha = getattr(args, "gsp_zipf_alpha", None) # Defensive validation for in-process callers that construct a # Namespace by hand and bypass the argparse boundary in