Function parse_one_bwa

examples/utils.py:263–274

(row)

        bwa = ast.literal_eval(bwa)  # parse again for string

    def parse_one_bwa(row):
        assert isinstance(row, list), f"Beam width array must be a list."
        assert len(
            row
        ) <= 8, "Length of beam width array must not be greater than 8 now."
        assert all([isinstance(beam, int) for beam in row
                    ]), "Numbers in beam width array must be integer."
        bwa_tensor = torch.zeros([8], dtype=torch.int32)
        for j in range(len(row)):
            bwa_tensor[j] = row[j]
        bwa_tensor[len(row):] = row[-1]
        return bwa_tensor, max(row)

    if isinstance(bwa, list):  # Only one BWA
        bwa_tensor, max_beam_width = parse_one_bwa(bwa)