[AMD] Fix AttributeError in GeneratedSharedPrefixDataset.from_args for in-process callers (#27580)

2026-06-08 21:53:14 -07:00
parent f6e6394d86
commit a32aeb688a
1 changed files with 2 additions and 2 deletions
@@ -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