Coverage for docs_src/events/tutorial003.py: 100%

1from contextlib import asynccontextmanager 6 ctx1abcdef

2 

3from fastapi import FastAPI 6 ctx1abcdef

4 

5 

6def fake_answer_to_everything_ml_model(x: float): 6 ctx1abcdef

7 return x * 42 6 ctx1ghijkl

8 

9 

10ml_models = {} 6 ctx1abcdef

11 

12 

13@asynccontextmanager 6 ctx1abcdef

14async def lifespan(app: FastAPI): 6 ctx1abcdef

15 # Load the ML model 

16 ml_models["answer_to_everything"] = fake_answer_to_everything_ml_model 12 ctx1gmhniojpkqlr

17 yield 12 ctx1gmhniojpkqlr

18 # Clean up the ML models and release the resources 

19 ml_models.clear() 12 ctx1gmhniojpkqlr

20 

21 

22app = FastAPI(lifespan=lifespan) 6 ctx1abcdef

23 

24 

25@app.get("/predict") 6 ctx1abcdef

26async def predict(x: float): 6 ctx1abcdef

27 result = ml_models["answer_to_everything"](x) 6 ctx1ghijkl

28 return {"result": result} 6 ctx1ghijkl