def call()

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

• 11 lines of code
• 6 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) == 1

        if inputs.dtype == tf.int8 or inputs.dtype == tf.int16:
            inputs = tf.cast(inputs, tf.int32)
        encoder_output = self.encoder_obj(
            inputs, training=training, mask=mask
        )

        return {'encoder_output': encoder_output}