hyper_parameters={"alpha": "1.0"},
data=[
DataBindingDir(conn_name='wine',
local_path='mlflow/sklearn/wine/wine-quality.csv'),
],
resources=ResourceRequirements(requests=ResourceList(cpu="2024m",
memory="2024Mi"),
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"
],
ToolchainIntegrationClient(),
ToolchainIntegration(
id=ENTITY_ID,
spec=ToolchainIntegrationSpec(
default_image="mock-image",
entrypoint="default-entrypoint",
),
),
toolchain_integration.toolchain_integration,
'ToolchainIntegration',
)
PACKAGING = EntityTestData(
ModelPackagingClient(),
ModelPackaging(id=ENTITY_ID,
spec=ModelPackagingSpec(artifact_name='test-artifact-name',
integration_name='test'),
status=ModelPackagingStatus(state=SUCCEEDED_STATE, )),
packaging.packaging,
'ModelPackaging',
)
PACKAGING_INTEGRATION = EntityTestData(
PackagingIntegrationClient(),
PackagingIntegration(
id=ENTITY_ID,
spec=PackagingIntegrationSpec(default_image="odahu:image",
entrypoint="some_entrypoint"),
),
packaging_integration.packaging_integration,
'PackagingIntegration',
)