[diffusion] CI: auto-skip diffusion tests when required pipeline class is missing from diffusers (#21139)
This commit is contained in:
@@ -118,7 +118,20 @@ def diffusion_server(case: DiffusionTestCase) -> ServerContext:
|
|||||||
extra_args=extra_args,
|
extra_args=extra_args,
|
||||||
env_vars=env_vars,
|
env_vars=env_vars,
|
||||||
)
|
)
|
||||||
ctx = manager.start()
|
try:
|
||||||
|
ctx = manager.start()
|
||||||
|
except (RuntimeError, TimeoutError) as exc:
|
||||||
|
# Auto-skip when the installed diffusers version lacks the required
|
||||||
|
# pipeline class. This avoids hard failures when a model needs a
|
||||||
|
# newer diffusers release than what is currently installed in CI.
|
||||||
|
msg = str(exc)
|
||||||
|
if "not found in diffusers" in msg or "has no attribute" in msg:
|
||||||
|
pytest.skip(
|
||||||
|
f"Skipping {case.id}: required diffusers pipeline class "
|
||||||
|
f"is not available in the installed version. "
|
||||||
|
f"Upgrade diffusers to enable this test."
|
||||||
|
)
|
||||||
|
raise
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# Reconstruct output size for OpenAI API
|
# Reconstruct output size for OpenAI API
|
||||||
|
|||||||
Reference in New Issue
Block a user