Another small fix to RetinaNet training example (#1306)

* fix to retinanet example

* linting fix

Author: quantumalaviya

examples/training/object_detection/pascal_voc/retina_net.py

@@ -227,7 +227,7 @@ def apply(inputs):
         bounding_boxes = keras_cv.bounding_box.convert_format(
             boxes, images=image, source="yxyx", target=bounding_box_format
         )
-        bounding_boxes = {"boxes": boxes, "classes": classes}
+        bounding_boxes = {"boxes": bounding_boxes, "classes": classes}
         return image, bounding_boxes
 
     return apply
@@ -278,10 +278,13 @@ def apply(inputs):
     return apply
 
 
-def pad_fn(images, boxes):
-    boxes = boxes.to_tensor(default_value=-1.0, shape=[GLOBAL_BATCH_SIZE, 32, 5])
-    boxes = boxes[:, :, :4]
-    classes = boxes[:, :, 4]
+def pad_fn(images, bounding_boxes):
+    boxes = bounding_boxes["boxes"].to_tensor(
+        default_value=-1.0, shape=[GLOBAL_BATCH_SIZE, 32, 4]
+    )
+    classes = bounding_boxes["classes"].to_tensor(
+        default_value=-1.0, shape=[GLOBAL_BATCH_SIZE, 32]
+    )
     return images, {"boxes": boxes, "classes": classes}