codeflash-admin/codeflash-internal
1
0
Fork 0
mirror of https://github.com/codeflash-ai/codeflash-internal.git synced 2026-05-04 18:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

revert to the previous code

Browse source
This commit is contained in:
Saurabh Misra 2024-11-01 16:53:56 -07:00
parent 5c7f6cd4b9
commit 6d66f17f80
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cli/codeflash/verification/test_results.py
View file

@ -84,10 +84,7 @@ class TestResults(BaseModel):
self.test_results.extend(other.test_results)
def get_by_id(self, invocation_id: InvocationId) -> FunctionTestInvocation | None:
for result in self.test_results:
if result.id == invocation_id:
return result
return None
return next((r for r in self.test_results if r.id == invocation_id), None)
def get_all_ids(self) -> set[InvocationId]:
return {test_result.id for test_result in self.test_results}