mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-10-04 21:33:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Accept native false Params values in live model placement check

Browse Source
This commit is contained in:
firestar5683
2026-09-19 21:53:52 -07:00
parent b108b958bd
commit 31be93b22c
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
roadscore/prototype/live_health.py
+1 -1
View File
@@ -25,7 +25,7 @@ def identity(value):
def local_model_placement(metadata, active, loading):
return metadata.get("uses_external_gpu") is False and active in (b"0", "0") and loading in (b"0", "0")
return metadata.get("uses_external_gpu") is False and (active is False or active in (b"0", "0")) and (loading is False or loading in (b"0", "0"))
def evaluate(snapshot, record, now):
roadscore/prototype/test_live_health.py
+2
View File
@@ -23,6 +23,8 @@ def authorization():
class Tests(unittest.TestCase):
def test_model_placement_requires_actual_runtime_flags(self):
self.assertTrue(local_model_placement({"uses_external_gpu":False},b"0",b"0"))
self.assertTrue(local_model_placement({"uses_external_gpu":False},False,False))
self.assertFalse(local_model_placement({"uses_external_gpu":False},None,False))
for active,loading in ((None,b"0"),(b"1",b"0"),(b"0",None),(b"0",b"1")):
self.assertFalse(local_model_placement({"uses_external_gpu":False},active,loading))
self.assertFalse(local_model_placement({},b"0",b"0"))