def test_validate_pred_label_valid(self) -> None:
        """Test validation of valid prediction labels."""
        labels = torch.tensor([[1], [0], [1], [1]])
        validated_labels = self.validator.validate_pred_label(labels)
        assert torch.equal(validated_labels, torch.tensor([True, False, True, True]))
Test validation of valid prediction labels.
tests/unit/data/validators/torch/test_depth.py