def predict(self, data: torch.Tensor) -> torch.Tensor:
        """Predict cluster assignments for the input data.

        Args:
            data (torch.Tensor): Input samples.
                Shape: ``(n_samples, n_features)``.

        Returns:
            torch.Tensor: Predicted cluster labels.
                Shape: ``(n_samples,)``.
        """
        _, resp = self._e_step(data)
        return torch.argmax(resp, axis=1)
Predict cluster assignments for the input data.

        Args:
            data (torch.Tensor): Input samples.
                Shape: ``(n_samples, n_features)``.

        Returns:
            torch.Tensor: Predicted cluster labels.
                Shape: ``(n_samples,)``.
        
src/anomalib/models/components/cluster/gmm.py