def edit(self, md: ModelDeployment) -> ModelDeployment:
"""
Edit Model Deployment
:param md: Model Deployment
:return Message from API server
"""
return ModelDeployment.from_dict(
self.query(MODEL_DEPLOYMENT_URL, action='PUT', payload=md.to_dict())
)
async def create(self, md: ModelDeployment) -> ModelDeployment:
"""
Create Model Deployment
:param md: Model Deployment
:return Message from API server
"""
return ModelDeployment.from_dict(
await self.query(MODEL_DEPLOYMENT_URL, action='POST', payload=md.to_dict())
)
async def get(self, name: str) -> ModelDeployment:
"""
Get Model Deployment from API server
:param name: Model Deployment name
:return: Model Deployment
"""
return ModelDeployment.from_dict(await self.query(f'{MODEL_DEPLOYMENT_URL}/{name}'))
def get_all(self, labels: typing.Dict[str, str] = None) -> typing.List[ModelDeployment]:
"""
Get all Model Deployments from API server
:return: all Model Deployments
"""
if labels:
url = f'{MODEL_DEPLOYMENT_URL}?{parse.urlencode(labels)}'
else:
url = MODEL_DEPLOYMENT_URL
return [ModelDeployment.from_dict(md) for md in self.query(url)]
limits=ResourceList(cpu="2024m",
memory="2024Mi")),
vcs_name="odahu-flow-examples"),
)
packaging_id = "airlfow-wine"
packaging = ModelPackaging(
id=packaging_id,
spec=ModelPackagingSpec(
targets=[Target(name="docker-push", connection_name="docker-ci")],
integration_name="docker-rest"),
)
deployment_id = "airlfow-wine"
deployment = ModelDeployment(
id=deployment_id,
spec=ModelDeploymentSpec(min_replicas=1, ),
)
model_example_request = {
"columns": [
"alcohol", "chlorides", "citric acid", "density", "fixed acidity",
"free sulfur dioxide", "pH", "residual sugar", "sulphates",
"total sulfur dioxide", "volatile acidity"
],
"data": [[12.8, 0.029, 0.48, 0.98, 6.2, 29, 3.33, 1.2, 0.39, 75, 0.66],
[12.8, 0.029, 0.48, 0.98, 6.2, 29, 3.33, 1.2, 0.39, 75, 0.66]]
}
dag = DAG('airlfow-wine', default_args=default_args, schedule_interval=None)
with dag:
PackagingIntegrationClient(),
PackagingIntegration(
id=ENTITY_ID,
spec=PackagingIntegrationSpec(default_image="odahu:image",
entrypoint="some_entrypoint"),
),
packaging_integration.packaging_integration,
'PackagingIntegration',
)
DEPLOYMENT = EntityTestData(
ModelDeploymentClient(),
ModelDeployment(id=ENTITY_ID,
spec=ModelDeploymentSpec(image="odahu:image",
min_replicas=0),
status=ModelDeploymentStatus(
state=READY_STATE,
available_replicas=1,
)),
deployment.deployment,
'ModelDeployment',
)
ROUTER = EntityTestData(
ModelRouteClient(),
ModelRoute(id=ENTITY_ID,
spec=ModelRouteSpec(mirror="test", ),
status=ModelRouteStatus(state=READY_STATE, )),
route.route,
'ModelRoute',
)