def call()

in code/source/custom_layer.py [0:0]

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

    def call(self, model_inputs):
        inputs = [K.cast(x, dtype="int32") for x in model_inputs]
        input_word_ids, input_mask, segment_ids = inputs
        bert_inputs = dict(
            input_word_ids=input_word_ids, input_mask=input_mask, input_type_ids=segment_ids
        )

        embed_output = self.bert_layer(inputs=bert_inputs)["sequence_output"]
        
        return embed_output