From 462d323509b2624e6ac33d304dcaf4f2e3532b6b Mon Sep 17 00:00:00 2001 From: Raffi Khatchadourian Date: Wed, 22 Nov 2023 12:40:49 -0500 Subject: [PATCH] Let's make the two tests only diff in the invocation. --- com.ibm.wala.cast.python.test/data/tf2_test_model_call.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/com.ibm.wala.cast.python.test/data/tf2_test_model_call.py b/com.ibm.wala.cast.python.test/data/tf2_test_model_call.py index 466b01491..e65cbbd69 100644 --- a/com.ibm.wala.cast.python.test/data/tf2_test_model_call.py +++ b/com.ibm.wala.cast.python.test/data/tf2_test_model_call.py @@ -1,7 +1,8 @@ import tensorflow as tf - # Create an override model to classify pictures + + class SequentialModel(tf.keras.Model): def __init__(self, **kwargs): @@ -28,6 +29,7 @@ def __call__(self, x): return x + input_data = tf.random.uniform([20, 28, 28]) model = SequentialModel()