嵌套模型
您需要为一个批处理任务创建架构:该任务会对一组输入运行天气预测模型,并将所有结果以列表形式返回。
本练习是课程的一部分
使用 FastAPI 在生产环境中部署 AI
练习说明
- 从
typing导入可用于嵌套列表的类。 - 使用
BaseModel创建类BatchInput。 - 在
BatchInput模型中,完善inputs字段,使其为ModelInput架构的列表。
交互式实操练习
通过完成这段示例代码来试试这个练习。
from datetime import date
from pydantic import BaseModel
# Import class for nested lists
from typing import ____
class ModelInput(BaseModel):
latitude: float
longitude: float
date: date
# Create batch input model
class ____(BaseModel):
job_name: str
# Inputs are list of model inputs
inputs: ___[___]