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

1from contextlib import asynccontextmanager 7 ctx1abcdefg

2 

3from fastapi import FastAPI 7 ctx1abcdefg

4 

5 

6def fake_answer_to_everything_ml_model(x: float): 7 ctx1abcdefg

7 return x * 42 7 ctx1hijklmn

8 

9 

10ml_models = {} 7 ctx1abcdefg

11 

12 

13@asynccontextmanager 7 ctx1abcdefg

14async def lifespan(app: FastAPI): 7 ctx1abcdefg

15 # Load the ML model 

16 ml_models["answer_to_everything"] = fake_answer_to_everything_ml_model 14 ctx1hoipjqkrlsmtnu

17 yield 14 ctx1hoipjqkrlsmtnu

18 # Clean up the ML models and release the resources 

19 ml_models.clear() 14 ctx1hoipjqkrlsmtnu

20 

21 

22app = FastAPI(lifespan=lifespan) 7 ctx1abcdefg

23 

24 

25@app.get("/predict") 7 ctx1abcdefg

26async def predict(x: float): 7 ctx1abcdefg

27 result = ml_models["answer_to_everything"](x) 7 ctx1hijklmn

28 return {"result": result} 7 ctx1hijklmn