Skip to content

Commit

Permalink
Let's make the two tests only diff in the invocation.
Browse files Browse the repository at this point in the history
  • Loading branch information
khatchad committed Nov 22, 2023
1 parent 4582417 commit 462d323
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion com.ibm.wala.cast.python.test/data/tf2_test_model_call.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
import tensorflow as tf


# Create an override model to classify pictures


class SequentialModel(tf.keras.Model):

def __init__(self, **kwargs):
Expand All @@ -28,6 +29,7 @@ def __call__(self, x):

return x


input_data = tf.random.uniform([20, 28, 28])

model = SequentialModel()
Expand Down

0 comments on commit 462d323

Please sign in to comment.