def __init__()

in optimum/furiosa/quantization.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

    def __init__(self, dataset: Dataset, input_datatypes, batch_size: int = 1):
        if dataset is None:
            raise ValueError("Provided dataset is None.")

        if input_datatypes is None:
            raise ValueError("Provided input_datatypes is None.")

        if batch_size <= 0:
            raise ValueError(f"Provided batch_size should be >= 1 (got: {batch_size}).")

        self.dataset = dataset
        self.input_datatypes = input_datatypes
        self.batch_size = batch_size

        self._dataset_iter = iter(self.dataset)