github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-03 17:39:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix pylint

Browse Source
This commit is contained in:
FangYin Cheng 2023-09-11 17:28:22 +08:00
parent 986ada3aeb
commit 27d7f9f6b9
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pilot/model/cluster/controller/tests/test_registry.py
View File

@ -88,12 +88,7 @@ async def test_send_heartbeat(model_registry, model_instance):
await model_registry.register_instance(model_instance) await model_registry.register_instance(model_instance)
last_heartbeat = datetime.now() - timedelta(seconds=10) last_heartbeat = datetime.now() - timedelta(seconds=10)
model_instance.last_heartbeat = last_heartbeat model_instance.last_heartbeat = last_heartbeat
assert ( assert await model_registry.send_heartbeat(model_instance) == True
await model_registry.send_heartbeat(
model_instance
)
== True
)
assert ( assert (
model_registry.registry[model_instance.model_name][0].last_heartbeat model_registry.registry[model_instance.model_name][0].last_heartbeat
> last_heartbeat > last_heartbeat