diff --git a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java index 6399a175a..c50bdfce4 100644 --- a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java +++ b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java @@ -202,6 +202,7 @@ public void testTf2() 0); // NOTE: Change to testTf2("tf2_test_dataset.py", "add", 2, 3, 2, 3) once // https://github.com/wala/ML/issues/89 is fixed. testTf2("tf2_test_tensor_list.py", "add", 2, 3, 2, 3); + testTf2("tf2_test_tensor_list2.py", "add", 0, 2); } private void testTf2( diff --git a/com.ibm.wala.cast.python.test/data/tf2_test_tensor_list2.py b/com.ibm.wala.cast.python.test/data/tf2_test_tensor_list2.py new file mode 100644 index 000000000..33e191933 --- /dev/null +++ b/com.ibm.wala.cast.python.test/data/tf2_test_tensor_list2.py @@ -0,0 +1,10 @@ +import tensorflow as tf + + +def add(a, b): + return a + b + + +list = [tf.ones([1, 2]), tf.ones([2, 2])] + +c = add(list, list)