Add correct params, but test will fail

ponder-lab · Nov 30, 2023 · 5e29516 · 5e29516
1 parent 72610a6
commit 5e29516
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java b/...wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java
@@ -217,9 +217,9 @@ public void testTf2()
     testTf2("tf2_test_model_call3.py", "SequentialModel.call", 1, 4, 2);
     testTf2("tf2_test_model_call4.py", "SequentialModel.__call__", 1, 4, 2);
     testTf2("tf2_test_callbacks.py", "get_dataset", 0, 0);
-    testTf2("tf2_test_callbacks2.py", "replica_fn", 0, 2);
+    testTf2("tf2_test_callbacks2.py", "replica_fn", 1, 2, 2);
     testTf2("tf2_test_callbacks3.py", "get_dataset", 0, 0);
-    testTf2("tf2_test_callbacks4.py", "replica_fn", 0, 2);
+    testTf2("tf2_test_callbacks4.py", "replica_fn", 1, 2, 2);
   }
 
   private void testTf2(