e164c3574da160fb72fc962947a783fa299424e3,sonnet/src/batch_apply_test.py,,,#,66

Before Change

    assert len(x.shape) == 2, "Requires rank 2 input."
    return x + 1.

if __name__ == "__main__":
  // tf.enable_v2_behavior()
  tf.test.main()

After Change

from sonnet.src import test_utils
import tensorflow as tf

EXAMPLE_INPUTS = (
    ((1, 2, 3), 1),
    ((1, 2, 3), 2),
    ((1, 2, 3, 4), 3),
    ((1, 2, 3, 4, 5, 6), 4),
)


class BatchApplyTest(test_utils.TestCase):

Instances