Fix spotless check

tensorflow · May 11, 2024 · 5cf1568 · 5cf1568
1 parent cac7ef7
commit 5cf1568
Showing 1 changed file with 2 additions and 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
@@ -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());