[BugFix][CI/Build][ROCM] Fix import error and apply assert in appropriate case in test_struct_output_generate (#28311)

Signed-off-by: Randall Smith <ransmith@amd.com>
Co-authored-by: Randall Smith <ransmith@amd.com>
This commit is contained in:
rasmith 2025-11-14 00:41:29 -06:00 committed by GitHub
parent 15ae8e0784
commit 93103575ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -677,9 +677,14 @@ def test_structured_output_with_reasoning_matrices(
reasoning, content = run_reasoning_extraction(reasoner, [generated_text]) reasoning, content = run_reasoning_extraction(reasoner, [generated_text])
print(f"Prompt: {prompt!r}\nReasoning: {reasoning!r}\nContent: {content!r}") print(f"Prompt: {prompt!r}\nReasoning: {reasoning!r}\nContent: {content!r}")
assert content is not None and reasoning is not None if "Qwen3" in model_name:
output_json = json.loads(content) assert content is not None
jsonschema.validate(instance=output_json, schema=reasoning_schema)
assert reasoning is not None
if content is not None:
output_json = json.loads(content)
jsonschema.validate(instance=output_json, schema=reasoning_schema)
@pytest.mark.skip_global_cleanup @pytest.mark.skip_global_cleanup