tfx.components.infra_validator.serving_bins.ServingBinary

View source on GitHub

Base class for serving binaries.

Attributes:

• container_port: Container port of the model server.

  Only applies to docker compatible serving binaries.

• image: Container image of the model server.

  Only applies to docker compatible serving binaries.

Methods

MakeClient

View source

MakeClient(
    endpoint
)

Create a model server client of this serving binary.

MakeDockerRunParams

View source

MakeDockerRunParams(
    *args
)

Make parameters for docker client.containers.run.

Only applies to docker compatible serving binaries.

Args:

• *args: List of unresolved variables to configure docker run parameters.

Returns:

A dictionary of docker run parameters.

MakeEnvVars

View source

MakeEnvVars(
    *args
)

Construct environment variables to be used in container image.

Only applies to docker compatible serving binaries.

Args:

• *args: List of unresolved variables to configure environment variables.

Returns:

A dictionary of environment variables inside container.