Fix type error.

Change-Id: I5db22405c93b66fd268b19f1e081dc24f3246d24

tests/test_models.py

@@ -166,7 +166,7 @@ def test_max_temperature(self):
  self.responses = {
  "get_model": protos.Model(name=name, max_temperature=max_temperature),
  }
- model = models.get_model(name)
+ model = models.get_base_model(name)
  self.assertEqual(max_temperature, model.max_temperature)
 
  @parameterized.named_parameters(