add branch name to results

23158a42ad5b88bc9e9d47a5c134fa90783b738a · 2024-12-11 12:59:55 +0000 · Alex Cheema

Files touched

M .github/bench.py

Diff

commit 23158a42ad5b88bc9e9d47a5c134fa90783b738a
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Wed Dec 11 12:59:55 2024 +0000

    add branch name to results
---
 .github/bench.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/bench.py b/.github/bench.py
index 77d1f555..f26dc4fb 100644
--- a/.github/bench.py
+++ b/.github/bench.py
@@ -23,6 +23,7 @@ async def measure_performance(api_endpoint: str, prompt: str, model: str) -> Dic
     results = {
         'model': model,
         'run_id': os.environ.get('GITHUB_RUN_ID', 'unknown'),
+        'branch': os.environ.get('GITHUB_REF_NAME', 'unknown'),
         'configuration': json.loads(os.environ.get('HARDWARE_CONFIG', '{}'))
     }

← 18e79199 test 30 · back to Exo · check gpu usage afe71c01 →