surface_code_in_stem/decoders/__init__.py

"""Decoder interfaces and adapters for surface-code experiments."""

from .base import DecoderInput, DecoderMetadata, DecoderOutput, DecoderProtocol
from .cuda_q_decoder import CudaQDecoder, CuQNNBackendAdapterDecoder, QuJaxNeuralBPDecoder
from .ising_predecoder import IsingDecoder
from .mwpm import MWPMDecoder
from .sparse_blossom import SparseBlossomDecoder
from .union_find import UnionFindDecoder, ConfidenceAwareUnionFindDecoder


class JAXConfidenceDecoderAdapter(ConfidenceAwareUnionFindDecoder):
    """Protocol-compatible fallback for the optional JAX confidence decoder."""

    def __init__(self, name: str = "jax_confidence", confidence_scale: float = 1.0, timeout_ms: float = 1000.0):
        super().__init__(name=name, confidence_scale=confidence_scale, timeout_ms=timeout_ms)

__all__ = [
    "DecoderInput",
    "DecoderMetadata",
    "DecoderOutput",
    "DecoderProtocol",
    "CudaQDecoder",
    "CuQNNBackendAdapterDecoder",
    "QuJaxNeuralBPDecoder",
    "MWPMDecoder",
    "UnionFindDecoder",
    "ConfidenceAwareUnionFindDecoder",
    "JAXConfidenceDecoderAdapter",
    "IsingDecoder",
    "SparseBlossomDecoder",
]