Skip to content

Commit bafbb3c

Browse files
committed
test fix for rebase
Signed-off-by: Ryan Nett <[email protected]>
1 parent ab6cb65 commit bafbb3c

File tree

1 file changed

+3
-3
lines changed
  • tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core

1 file changed

+3
-3
lines changed

tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/IndexingTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -61,11 +61,11 @@ public void testStridedSliceIndex(){
6161
Session sess = new Session(g)) {
6262
Scope scope = new Scope(g);
6363
long[] shape = {10, 10, 10, 10, 10, 10, 10, 10};
64-
Zeros<TFloat32> op = Zeros.create(scope, Constant.vectorOf(scope, shape), TFloat32.DTYPE);
64+
Zeros<TFloat32> op = Zeros.create(scope, Constant.vectorOf(scope, shape), TFloat32.class);
6565
StridedSlice<TFloat32> output = StridedSliceHelper.stridedSlice(scope, op, slice);
66-
try (Tensor<TFloat32> result = sess.runner().fetch(output.asOutput()).run().get(0).expect(TFloat32.DTYPE)) {
66+
try (TFloat32 result = (TFloat32) sess.runner().fetch(output.asOutput()).run().get(0)) {
6767
// expected shape from Python tensorflow
68-
assertEquals(Shape.of(1, 10, 1, 10, 10, 10, 4, 3), result.data().shape(), "Slice index didn't match expected (Python)");
68+
assertEquals(Shape.of(1, 10, 1, 10, 10, 10, 4, 3), result.shape(), "Slice index didn't match expected (Python)");
6969
}
7070
}
7171
}

0 commit comments

Comments
 (0)