Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfx.components.infra_validator.model_server_clients.base_client.BaseModelServerClient

View source on GitHub

Common interface for all model server clients.

Methods

SendRequests

View source

SendRequests(
    requests
)

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, polling_interval_sec
)

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.