def METHOD_NAME(self):
    with self.assertRaises(RuntimeError):
        # SGD unsupported currently
        testing_utils.train_model(
            dict(
                task='integration_tests:candidate',
                model='transformer/ranker',
                optimizer='sgd',
                fp16=True,
                fp16_impl='mem_efficient',
            )
        )