sagemaker.serve.utils.task#
Accessors to retrieve task fallback input/output schema
Functions
|
Retrieves task sample inputs and outputs locally. |
- sagemaker.serve.utils.task.retrieve_local_schemas(task: str) Tuple[Any, Any][source]#
Retrieves task sample inputs and outputs locally.
- Parameters:
task (str) – Required, the task name
- Returns:
A tuple that contains the sample input, at index 0, and output schema, at index 1.
- Return type:
Tuple[Any, Any]
- Raises:
ValueError – If no tasks config found or the task does not exist in the local config.