def call()

in ludwig/features/text_feature.py [0:0]

• 15 lines of code
• 5 McCabe index (conditional complexity)

    def call(self, inputs, training=None, mask=None):
        assert isinstance(inputs, tf.Tensor)
        assert inputs.dtype == tf.int8 or inputs.dtype == tf.int16 or \
               inputs.dtype == tf.int32 or inputs.dtype == tf.int64
        assert len(inputs.shape) == 2

        inputs_exp = tf.cast(inputs, dtype=tf.int32)

        if self.pad_idx is not None:
            inputs_mask = tf.cast(tf.not_equal(inputs, self.pad_idx),
                                dtype=tf.int32)
        else: 
            inputs_mask = None

        encoder_output = self.encoder_obj(
            inputs_exp, training=training, mask=inputs_mask
        )

        return encoder_output