amend

hengtaoguo · hengtaoguo · commit c94b9faede98 · 2026-03-31T05:25:04.000Z
diff --git a/tests/utils/forward_pass_logit_checker.py b/tests/utils/forward_pass_logit_checker.py
@@ -543,6 +543,7 @@ def main(config, test_args):  # pylint: disable=W0621
       choices=["linen", "nnx"],
       help="Checkpoint format to load: 'linen' (default) or 'nnx'.",
   )
+
   # Parse known args returns the namespace AND the list of remaining arguments
   test_args, remaining_args = parser.parse_known_args()
   # Reconstruct model_args (script name + the args MaxText needs)

Original file line number	Diff line number	Diff line change
`@@ -543,6 +543,7 @@ def main(config, test_args): # pylint: disable=W0621`
`543`	`543`	`choices=["linen", "nnx"],`
`544`	`544`	`help="Checkpoint format to load: 'linen' (default) or 'nnx'.",`
`545`	`545`	`)`
	`546`	`+`
`546`	`547`	`# Parse known args returns the namespace AND the list of remaining arguments`
`547`	`548`	`test_args, remaining_args = parser.parse_known_args()`
`548`	`549`	`# Reconstruct model_args (script name + the args MaxText needs)`