a0a14ed743f511932db8234b2521a9eebfecde31,tests/ludwig/models/modules/test_encoder.py,,encoder_test,#Any#Any#Any#Any#Any#Any#Any#,69

Before Change

    // Run the encoder
    input_data = tf.convert_to_tensor(input_data)
    dropout_rate = tf.convert_to_tensor(dropout_rate)
    is_training = tf.convert_to_tensor(False)

    hidden, _ = encoder(
        input_data,
        regularizer,
        dropout_rate,
        is_training=is_training
    )

    // Check output shape and type
    assert hidden.dtype == output_dtype
    assert hidden.shape.as_list() == output_shape

After Change

    // Run the encoder
    input_data = tf.convert_to_tensor(input_data)

    hidden = encoder(
        input_data,
        training=False
    )["encoder_output"]

    // Check output shape and type
    assert hidden.dtype == output_dtype
    assert hidden.shape.as_list() == output_shape

Instances