Inference test

Author: LukeWood

keras_cv/layers/preprocessing/jittered_resize.py

@@ -131,7 +131,7 @@ def call(self, inputs, training=True):
         else:
             inputs = self._ensure_inputs_are_compute_dtype(inputs)
             inputs, meta_data = self._format_inputs(inputs)
-            output = self.inference_resizing(inputs)
+            output = self._inference_resizing(inputs)
             return self._format_output(output, meta_data)
 
     def get_random_transformation(self, image=None, **kwargs):

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"],