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

@abc.abstractmethod
MakeClient(
    endpoint: Text
) -> tfx.components.infra_validator.model_server_clients.base_client.BaseModelServerClient

Create a model server client of this serving binary.

MakeDockerRunParams

View source

@abc.abstractmethod
MakeDockerRunParams(
    *args
) -> Dict[Text, Text]

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

@abc.abstractmethod
MakeEnvVars(
    *args
) -> Dict[Text, Text]

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.