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

1from contextlib import asynccontextmanager 4 ctx1abcd

2 

3from fastapi import FastAPI 4 ctx1abcd

4 

5 

6def fake_answer_to_everything_ml_model(x: float): 4 ctx1abcd

7 return x * 42 4 ctx1efgh

8 

9 

10ml_models = {} 4 ctx1abcd

11 

12 

13@asynccontextmanager 4 ctx1abcd

14async def lifespan(app: FastAPI): 4 ctx1abcd

15 # Load the ML model 

16 ml_models["answer_to_everything"] = fake_answer_to_everything_ml_model 8 ctx1eifjgkhl

17 yield 8 ctx1eifjgkhl

18 # Clean up the ML models and release the resources 

19 ml_models.clear() 8 ctx1eifjgkhl

20 

21 

22app = FastAPI(lifespan=lifespan) 4 ctx1abcd

23 

24 

25@app.get("/predict") 4 ctx1abcd

26async def predict(x: float): 4 ctx1abcd

27 result = ml_models["answer_to_everything"](x) 4 ctx1efgh

28 return {"result": result} 4 ctx1efgh