def enqueue_one_symbol()

in neuralcompression/entropy_coders/jax_arithemetic_coder.py [0:0]

• 25 lines of code
• 1 McCabe index (conditional complexity)

    def enqueue_one_symbol(msg_index, enqueue_state):
        symbol = message[msg_index]
        (
            int_array,
            byte,
            idx,
            bit_idx,
            pending_bits,
            high,
            low,
            cdf_state,
        ) = enqueue_state
        return _enqueue(
            symbol,
            int_array,
            byte,
            idx,
            bit_idx,
            pending_bits,
            high,
            low,
            cdf_fun,
            cdf_state,
            precision,
        )