修复类型错误

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

@@ -18,7 +18,7 @@ async def fastapi_add_valve(
     v_type: str = VALVES_TYPE_PRV,
     setting: float = 0,
     minor_loss: float = 0,
-) -> ChangeSet:
+) -> "ChangeSet":
     ps = {
         "id": valve,
         "node1": node1,
@@ -32,7 +32,7 @@ async def fastapi_add_valve(
     return add_valve(network, ChangeSet(ps))
 
 @router.post("/deletevalve/", response_model=None)
-async def fastapi_delete_valve(network: str, valve: str) -> ChangeSet:
+async def fastapi_delete_valve(network: str, valve: str) -> "ChangeSet":
     ps = {"id": valve}
     return delete_valve(network, ChangeSet(ps))
 
@@ -67,31 +67,31 @@ async def fastapi_get_valve_minor_loss(network: str, valve: str) -> float | None
     return ps["minor_loss"]
 
 @router.post("/setvalvenode1/", response_model=None)
-async def fastapi_set_valve_node1(network: str, valve: str, node1: str) -> ChangeSet:
+async def fastapi_set_valve_node1(network: str, valve: str, node1: str) -> "ChangeSet":
     ps = {"id": valve, "node1": node1}
     return set_valve(network, ChangeSet(ps))
 
 @router.post("/setvalvenode2/", response_model=None)
-async def fastapi_set_valve_node2(network: str, valve: str, node2: str) -> ChangeSet:
+async def fastapi_set_valve_node2(network: str, valve: str, node2: str) -> "ChangeSet":
     ps = {"id": valve, "node2": node2}
     return set_valve(network, ChangeSet(ps))
 
 @router.post("/setvalvenodediameter/", response_model=None)
 async def fastapi_set_valve_diameter(
     network: str, valve: str, diameter: float
-) -> ChangeSet:
+) -> "ChangeSet":
     ps = {"id": valve, "diameter": diameter}
     return set_valve(network, ChangeSet(ps))
 
 @router.post("/setvalvetype/", response_model=None)
-async def fastapi_set_valve_type(network: str, valve: str, type: str) -> ChangeSet:
+async def fastapi_set_valve_type(network: str, valve: str, type: str) -> "ChangeSet":
     ps = {"id": valve, "type": type}
     return set_valve(network, ChangeSet(ps))
 
 @router.post("/setvalvesetting/", response_model=None)
 async def fastapi_set_valve_setting(
     network: str, valve: str, setting: float
-) -> ChangeSet:
+) -> "ChangeSet":
     ps = {"id": valve, "setting": setting}
     return set_valve(network, ChangeSet(ps))
 
@@ -109,7 +109,7 @@ async def fastapi_get_all_valve_properties(network: str) -> list[dict[str, Any]]
 @router.post("/setvalveproperties/", response_model=None)
 async def fastapi_set_valve_properties(
     network: str, valve: str, req: Request
-) -> ChangeSet:
+) -> "ChangeSet":
     props = await req.json()
     ps = {"id": valve} | props
     return set_valve(network, ChangeSet(ps))