Fix return type mismatch

This commit is contained in:
wqy
2022-09-17 22:34:27 +08:00
parent 2aa967546a
commit a645195eb5
3 changed files with 8 additions and 8 deletions

View File

@@ -47,12 +47,12 @@ def delete_pipe(name: str, id: str) -> ChangeSet:
def get_pipe_node1(name: str, id: str) -> str | None:
row = _get_pipe(name, id)
return float(row['node1']) if row != None else None
return row['node1'] if row != None else None
def get_pipe_node2(name: str, id: str) -> str | None:
row = _get_pipe(name, id)
return float(row['node2']) if row != None else None
return row['node2'] if row != None else None
def get_pipe_length(name: str, id: str) -> float | None:
@@ -77,7 +77,7 @@ def get_pipe_minor_loss(name: str, id: str) -> float | None:
def get_pipe_status(name: str, id: str) -> str | None:
row = _get_pipe(name, id)
return float(row['status']) if row != None else None
return row['status'] if row != None else None
def _set_pipe(name: str, id: str, key: str, key_type: str, value: str, optional: bool = False) -> ChangeSet: