def __init__()

in lmgvp/datasets.py [0:0]

• 26 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, data_list, **kwargs):
        """
        Initializes the dataset

        Args:
            data_list: List containint the initial data

        Returns:
            None
        """
        self.letter_to_num = {
            "C": 4,
            "D": 3,
            "S": 15,
            "Q": 5,
            "K": 11,
            "I": 9,
            "P": 14,
            "T": 16,
            "F": 13,
            "A": 0,
            "G": 7,
            "H": 8,
            "E": 6,
            "L": 10,
            "R": 1,
            "W": 17,
            "V": 19,
            "N": 2,
            "Y": 18,
            "M": 12,
            "X": 0,
        }
        self.num_to_letter = {v: k for k, v in self.letter_to_num.items()}
        super(StandardProteinGraphDataset, self).__init__(data_list, **kwargs)