from typing import Callable, Protocol

import numpy as np
import numpy.typing as npt

CSICallback = Callable[[npt.NDArray[np.complex64]], None]


class CSIProducer(Protocol):
    def __call__(self, csi_callback: CSICallback | None = None) -> None: ...