浏览代码

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