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()