修复类型错误

app/api/v1/endpoints/network/pumps.py

@@ -11,12 +11,12 @@ async def fastapi_get_pump_schema(network: str) -> dict[str, dict[str, Any]]:
 @router.post("/addpump/", response_model=None)
 async def fastapi_add_pump(
     network: str, pump: str, node1: str, node2: str, power: float = 0.0
-) -> ChangeSet:
+) -> "ChangeSet":
     ps = {"id": pump, "node1": node1, "node2": node2, "power": power}
     return add_pump(network, ChangeSet(ps))
 
 @router.post("/deletepump/", response_model=None)
-async def fastapi_delete_pump(network: str, pump: str) -> ChangeSet:
+async def fastapi_delete_pump(network: str, pump: str) -> "ChangeSet":
     ps = {"id": pump}
     return delete_pump(network, ChangeSet(ps))
 
@@ -31,12 +31,12 @@ async def fastapi_get_pump_node2(network: str, pump: str) -> str | None:
     return ps["node2"]
 
 @router.post("/setpumpnode1/", response_model=None)
-async def fastapi_set_pump_node1(network: str, pump: str, node1: str) -> ChangeSet:
+async def fastapi_set_pump_node1(network: str, pump: str, node1: str) -> "ChangeSet":
     ps = {"id": pump, "node1": node1}
     return set_pump(network, ChangeSet(ps))
 
 @router.post("/setpumpnode2/", response_model=None)
-async def fastapi_set_pump_node2(network: str, pump: str, node2: str) -> ChangeSet:
+async def fastapi_set_pump_node2(network: str, pump: str, node2: str) -> "ChangeSet":
     ps = {"id": pump, "node2": node2}
     return set_pump(network, ChangeSet(ps))
 
@@ -54,7 +54,7 @@ async def fastapi_get_all_pump_properties(network: str) -> list[dict[str, Any]]:
 @router.post("/setpumpproperties/", response_model=None)
 async def fastapi_set_pump_properties(
     network: str, pump: str, req: Request
-) -> ChangeSet:
+) -> "ChangeSet":
     props = await req.json()
     ps = {"id": pump} | props
     return set_pump(network, ChangeSet(ps))