predictions = self.estimator.predict(predict_features)
print("predictions: {}".format(type(predictions)))
while True:
  try:
    result = next(predictions)
    predict_labels.append(result)
  except StopIteration:
    return predict_labels

TypeError: unsupported callable