Inference test

LukeWood · LukeWood · commit 1432e2dff5b2 · 2023-01-26T10:23:54.000-08:00
diff --git a/keras_cv/layers/preprocessing/jittered_resize_test.py b/keras_cv/layers/preprocessing/jittered_resize_test.py
@@ -175,7 +175,7 @@ def test_augment_inference_mode(self):
             seed=self.seed,
         )
         output = layer(input, training=False)
-        expected_output = layer._inference_resizing(output)
+        expected_output = layer.inference_resizing(output)
         self.assertAllClose(
             expected_output["bounding_boxes"]["boxes"],
             output["bounding_boxes"]["boxes"],

Original file line number	Diff line number	Diff line change
`@@ -175,7 +175,7 @@ def test_augment_inference_mode(self):`
`175`	`175`	`seed=self.seed,`
`176`	`176`	`)`
`177`	`177`	`output = layer(input, training=False)`
`178`		`- expected_output = layer._inference_resizing(output)`
	`178`	`+ expected_output = layer.inference_resizing(output)`
`179`	`179`	`self.assertAllClose(`
`180`	`180`	`expected_output["bounding_boxes"]["boxes"],`
`181`	`181`	`output["bounding_boxes"]["boxes"],`