def to_proto(self):
data = self.dict()
if data.get("user_config"):
data["user_config"] = cloudpickle.dumps(data["user_config"])
if data.get("autoscaling_config"):
data["autoscaling_config"] = AutoscalingConfigProto(
**data["autoscaling_config"])
return DeploymentConfigProto(**data)
def to_proto_bytes(self):
data = self.dict()
if data.get("user_config"):
data["user_config"] = pickle.dumps(data["user_config"])
if data.get("autoscaling_config"):
data["autoscaling_config"] = AutoscalingConfigProto(
**data["autoscaling_config"])
return DeploymentConfigProto(
is_cross_language=False,
deployment_language=DeploymentLanguage.PYTHON,
**data,
).SerializeToString()