From 5cf156810a86c9c48cc60606a974927219fbc30a Mon Sep 17 00:00:00 2001 From: Karl Lessard Date: Sat, 11 May 2024 13:54:50 -0400 Subject: [PATCH] Fix spotless check --- .../src/test/java/org/tensorflow/SignatureTest.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SignatureTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SignatureTest.java index 819ce473b20..28e97939aa2 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SignatureTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SignatureTest.java @@ -22,7 +22,6 @@ import org.tensorflow.ndarray.Shape; import org.tensorflow.op.Ops; import org.tensorflow.op.core.Placeholder; -import org.tensorflow.op.math.Sign; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; @@ -100,7 +99,8 @@ public void createTensorInfoFromOperandWithUnknownShape() { public void createTensorInfoFromOperandWithPartiallyUnknownShape() { try (Graph g = new Graph()) { var tf = Ops.create(g); - var placeholder = tf.placeholder(TInt32.class, Placeholder.shape(Shape.of(Shape.UNKNOWN_SIZE, 10))); + var shape = Shape.of(Shape.UNKNOWN_SIZE, 10); + var placeholder = tf.placeholder(TInt32.class, Placeholder.shape(shape)); var tensorInfo = Signature.Builder.toTensorInfo(placeholder.asOutput()); assertFalse(tensorInfo.getTensorShape().getUnknownRank()); assertEquals(2, tensorInfo.getTensorShape().getDimCount());