tfx.components.infra_validator.model_server_clients.base_client.BaseModelServerClient

Common interface for all model server clients.

Methods

SendRequests

View source

SendRequests(
    requests: List[tfx.components.infra_validator.types.Request]
) -> None

Send requests to the model server.

Args
requests	A list of request protos.

Raises
ValidationFailed	If error occurred while sending requests.

WaitUntilModelLoaded

View source

WaitUntilModelLoaded(
    deadline: float,
    polling_interval_sec: int
) -> None

Wait until model is loaded and available.

Args
deadline	A deadline time in UTC timestamp (in seconds).
polling_interval_sec	GetServingStatus() polling interval.

Raises
DeadlineExceeded	When deadline exceeded before model is ready.
ValidationFailed	If validation failed explicitly.