Skip to content

Commit 22a10db

Browse files
committed
Explain optional Diffusers test import
Signed-off-by: realAsma <akuriparambi@nvidia.com>
1 parent aaac495 commit 22a10db

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/unit/torch/export/test_unified_export_hf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ def test_resolve_export_dtype(configured_dtype, dtype, expected_dtype, warning_c
6161

6262

6363
def test_resolve_export_dtype_with_empty_diffusers_config():
64+
# Import locally so Diffusers stays optional during torch-only test collection.
6465
frozen_dict = pytest.importorskip("diffusers.configuration_utils").FrozenDict()
6566
model = torch.nn.Linear(1, 1)
6667
model.config = frozen_dict

0 commit comments

Comments
 (0)