def init_minibatch()

in Models/exprsynth/nagdecoder.py [0:0]

• 30 lines of code
• 16 McCabe index (conditional complexity)

    def init_minibatch(self, batch_data: Dict[str, Any]) -> None:
        total_edge_types = len(self.__expansion_labeled_edge_types) + len(self.__expansion_unlabeled_edge_types)

        batch_data['eg_node_offset'] = 0
        batch_data['next_step_target_node_id'] = [0 for _ in range(self.hyperparameters['eg_propagation_substeps'])]

        batch_data['eg_node_token_ids'] = []
        batch_data['eg_initial_node_ids'] = []
        batch_data['eg_sending_node_ids'] = [[[] for _ in range(total_edge_types)]
                                             for _ in range(self.hyperparameters['eg_propagation_substeps'])]
        batch_data['eg_edge_label_ids'] = [[[] for _ in range(len(self.__expansion_labeled_edge_types))]
                                           for _ in range(self.hyperparameters['eg_propagation_substeps'])]
        batch_data['eg_msg_target_node_ids'] = [[[] for _ in range(total_edge_types)]
                                                for _ in range(self.hyperparameters['eg_propagation_substeps'])]
        batch_data['eg_receiving_node_ids'] = [[] for _ in range(self.hyperparameters['eg_propagation_substeps'])]
        batch_data['eg_receiving_node_nums'] = [0 for _ in range(self.hyperparameters['eg_propagation_substeps'])]

        batch_data['eg_production_nodes'] = []
        if self.hyperparameters['eg_use_vars_for_production_choice']:
            batch_data['eg_prod_idx_offset'] = 0
            batch_data['eg_production_var_last_use_node_ids'] = []
            batch_data['eg_production_var_last_use_node_ids_target_ids'] = []
        batch_data['eg_production_node_choices'] = []
        if self.hyperparameters.get('eg_use_context_attention', False):
            batch_data['eg_production_to_context_id'] = []
        batch_data['eg_varproduction_nodes'] = []
        batch_data['eg_varproduction_options_nodes'] = []
        batch_data['eg_varproduction_options_mask'] = []
        batch_data['eg_varproduction_node_choices'] = []
        batch_data['eg_litproduction_nodes'] = {literal_kind: [] for literal_kind in LITERAL_NONTERMINALS}
        batch_data['eg_litproduction_node_choices'] = {literal_kind: [] for literal_kind in LITERAL_NONTERMINALS}
        batch_data['eg_litproduction_to_context_id'] = {literal_kind: [] for literal_kind in LITERAL_NONTERMINALS}
        batch_data['eg_litproduction_choice_normalizer'] = {literal_kind: [] for literal_kind in LITERAL_NONTERMINALS}