biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

mypy: check additional directories (#9162)

Browse Source 
Signed-off-by: Russell Bryant <rbryant@redhat.com>
This commit is contained in:
Russell Bryant
2024-10-08 18:08:22 -04:00
committed by GitHub
parent bd37b9fbe2
commit 2a131965a8
4 changed files with 41 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/usage/usage_lib.py
View File

@@ -204,7 +204,7 @@ class UsageMessage:
self._write_to_file(data)
self._send_to_server(data)
def _send_to_server(self, data):
def _send_to_server(self, data: Dict[str, Any]) -> None:
try:
global_http_client = global_http_connection.get_sync_client()
global_http_client.post(_USAGE_STATS_SERVER, json=data)
@@ -212,7 +212,7 @@ class UsageMessage:
# silently ignore unless we are using debug log
logging.debug("Failed to send usage data to server")
def _write_to_file(self, data):
def _write_to_file(self, data: Dict[str, Any]) -> None:
os.makedirs(os.path.dirname(_USAGE_STATS_JSON_PATH), exist_ok=True)
Path(_USAGE_STATS_JSON_PATH).touch(exist_ok=True)
with open(_USAGE_STATS_JSON_PATH, "a") as f: