From 56361e2ec68b8aa06b2bcc45206492891a1e9ad8 Mon Sep 17 00:00:00 2001 From: Siarhei Fedartsou Date: Wed, 19 Jun 2024 18:02:08 +0200 Subject: [PATCH] wip --- scripts/ci/e2e_benchmark.py | 18 +++--------------- 1 file changed, 3 insertions(+), 15 deletions(-) diff --git a/scripts/ci/e2e_benchmark.py b/scripts/ci/e2e_benchmark.py index 8e2e51944..0366bb45d 100644 --- a/scripts/ci/e2e_benchmark.py +++ b/scripts/ci/e2e_benchmark.py @@ -10,9 +10,6 @@ import argparse from scipy import stats import logging -logging.getLogger("urllib3").setLevel(logging.CRITICAL) -logging.getLogger("requests").setLevel(logging.CRITICAL) - class BenchmarkRunner: def __init__(self, gps_traces_file_path): @@ -42,18 +39,9 @@ class BenchmarkRunner: response = requests.get(url) end_time = time.time() if response.status_code != 200: - if benchmark_name == 'match': - code = response.json()['code'] - if code == 'NoSegment' or code == 'NoMatch': - continue - elif benchmark_name == 'route': - code = response.json()['code'] - if code == 'NoRoute': - continue - elif benchmark_name == 'trip': - code = response.json()['code'] - if code == 'NoTrips': - continue + code = response.json()['code'] + if code in ['NoSegment', 'NoMatch', 'NoRoute', 'NoTrips']: + continue raise Exception(f"Error: {response.status_code} {response.text}") times.append((end_time - start_time) * 1000) # convert to ms