Skip to content

Commit 6dfa1a8

Browse files
committed
Update test.
1 parent c54739d commit 6dfa1a8

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1757,7 +1757,12 @@ public void testAdd104()
17571757
@Test
17581758
public void testAdd105()
17591759
throws ClassHierarchyException, IllegalArgumentException, CancelException, IOException {
1760-
test("tf2_test_add105.py", "add", 2, 2, Map.of(2, Set.of(MNIST_INPUT), 3, Set.of(MNIST_INPUT)));
1760+
test(
1761+
"tf2_test_add105.py",
1762+
"add",
1763+
2,
1764+
2,
1765+
Map.of(2, Set.of(TENSOR_4_FLOAT32), 3, Set.of(TENSOR_4_FLOAT32)));
17611766
}
17621767

17631768
@Test

0 commit comments

Comments
 (0)