def test_validate_pred_label_invalid_shape(self) -> None:
        """Test validation of prediction labels with invalid shape."""
        with pytest.raises(ValueError, match="Predicted label must be 1-dimensional or 2-dimensional"):
            self.validator.validate_pred_label(torch.tensor([[[1]], [[0]], [[1]], [[1]]]))
Test validation of prediction labels with invalid shape.
tests/unit/data/validators/torch/test_depth.py