def testWrongDeepLabVariant(self): model_options = common.ModelOptions([])._replace( model_variant='no_such_variant') with self.assertRaises(ValueError): model._get_logits(images=[], model_options=model_options)