Browse Source

add branch name to results

Alex Cheema 7 months ago
parent
commit
23158a42ad
1 changed files with 1 additions and 0 deletions
  1. 1 0
      .github/bench.py

+ 1 - 0
.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', '{}'))
     }