Skip to content

Commit ec4f679

Browse files
committed
added @SuppressWarnings("unchecked") on Variable array
1 parent 7915e63 commit ec4f679

File tree

1 file changed

+2
-0
lines changed
  • tensorflow-framework/src/test/java/org/tensorflow/framework/optimizers

1 file changed

+2
-0
lines changed

tensorflow-framework/src/test/java/org/tensorflow/framework/optimizers/AdamTest.java

+2
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,9 @@ public void testBasic() {
105105
Op update = instance.applyGradients(gradsAndVars, "AdamTest");
106106

107107
/* Create and validate the shapes of the slots */
108+
@SuppressWarnings("unchecked")
108109
Variable<TFloat32>[] firstMomentSlots = new Variable[2];
110+
@SuppressWarnings("unchecked")
109111
Variable<TFloat32>[] secondMomentSlots = new Variable[2];
110112

111113
firstMomentSlots[0] = instance.getSlot(var0.asOutput(), FIRST_MOMENT).get();

0 commit comments

Comments
 (0)