diff --git a/tests/datasets/test_substation.py b/tests/datasets/test_substation.py index 36acc44b548..30f2e7b42d2 100644 --- a/tests/datasets/test_substation.py +++ b/tests/datasets/test_substation.py @@ -89,7 +89,7 @@ def test_getitem_semantic(self, config: dict[str, Any]) -> None: assert isinstance(x, dict), f'Expected dict, got {type(x)}' assert isinstance(x['image'], torch.Tensor), ( 'Expected image to be a torch.Tensor' - ) + ) assert isinstance(x['mask'], torch.Tensor), 'Expected mask to be a torch.Tensor' def test_len(self, dataset: Substation) -> None: