def predict_step(
        self,
        batch: Batch,
        batch_idx: int,
        dataloader_idx: int = 0,
    ) -> STEP_OUTPUT:
        """Perform prediction step.

        This method is called during the predict stage of training. By default,
        it calls the validation step.

        Args:
            batch (Batch): Input batch
            batch_idx (int): Index of the batch
            dataloader_idx (int, optional): Index of the dataloader.
                Defaults to ``0``.

        Returns:
            STEP_OUTPUT: Model predictions
        """
        del dataloader_idx  # These variables are not used.

        return self.validation_step(batch, batch_idx)
Perform prediction step.

        This method is called during the predict stage of training. By default,
        it calls the validation step.

        Args:
            batch (Batch): Input batch
            batch_idx (int): Index of the batch
            dataloader_idx (int, optional): Index of the dataloader.
                Defaults to ``0``.

        Returns:
            STEP_OUTPUT: Model predictions
        
src/anomalib/models/components/base/anomalib_module.py