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 3 ctx1efg

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 6 ctx1ehfigj

17 yield 6 ctx1ehfigj

18 # Clean up the ML models and release the resources 

19 ml_models.clear() 6 ctx1ehfigj

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) 3 ctx1efg

28 return {"result": result} 3 ctx1efg