소스 검색

add a startup event for loading q learning model

chenhaiyang 4 년 전
부모
커밋
bbca349d37
1개의 변경된 파일14개의 추가작업 그리고 0개의 파일을 삭제
  1. 14 0
      app/core/events.py

+ 14 - 0
app/core/events.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+
+from typing import Callable, Optional
+
+from fastapi import FastAPI
+
+from app.controllers.events import load_q_learning_model
+
+
+def create_start_app_handler(app: Optional[FastAPI] = None) -> Callable:
+    async def start_app() -> None:
+        await load_q_learning_model()
+
+    return start_app