v1.2.1: Update the container labels in the integration tests (#1588)

We've recently changed the labels for the containers, as part of unifying the
training and inference containers. This meant that certain container verification
scripts were not getting run appropriately