From 91a8972e18fce55168584b66b6387475a630baf4 Mon Sep 17 00:00:00 2001
From: Hugo Larcher <hugo.larcher@huggingface.co>
Date: Thu, 18 Jul 2024 10:03:05 +0200
Subject: [PATCH] fix: Fix to allow report for a full failed test

---
 load_tests/parse_load_test.py |  9 +++++----
 load_tests/poetry.lock        | 14 +++++++-------
 load_tests/pyproject.toml     |  1 +
 3 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/load_tests/parse_load_test.py b/load_tests/parse_load_test.py
index 5caa61ed..e2b3c0cc 100644
--- a/load_tests/parse_load_test.py
+++ b/load_tests/parse_load_test.py
@@ -20,12 +20,13 @@ def get_nested(obj, path, default=None):
     for key in path.split("."):
         if obj is None:
             return default
-        if re.match(r"\d+", key):
-            key = int(key)
-            if key >= len(obj):
+        if isinstance(obj, list):
+            try:
+                key = int(key)
+            except ValueError:
                 return default
             obj = obj[key]
-        else:
+        elif isinstance(obj, dict):
             obj = obj.get(key, default)
     return obj
 
diff --git a/load_tests/poetry.lock b/load_tests/poetry.lock
index 81f7bb32..ab163f73 100644
--- a/load_tests/poetry.lock
+++ b/load_tests/poetry.lock
@@ -353,13 +353,13 @@ files = [
 
 [[package]]
 name = "huggingface-hub"
-version = "0.23.4"
+version = "0.23.5"
 description = "Client library to download and publish models, datasets and other repos on the huggingface.co hub"
 optional = false
 python-versions = ">=3.8.0"
 files = [
-    {file = "huggingface_hub-0.23.4-py3-none-any.whl", hash = "sha256:3a0b957aa87150addf0cc7bd71b4d954b78e749850e1e7fb29ebbd2db64ca037"},
-    {file = "huggingface_hub-0.23.4.tar.gz", hash = "sha256:35d99016433900e44ae7efe1c209164a5a81dbbcd53a52f99c281dcd7ce22431"},
+    {file = "huggingface_hub-0.23.5-py3-none-any.whl", hash = "sha256:d7a7d337615e11a45cc14a0ce5a605db6b038dc24af42866f731684825226e90"},
+    {file = "huggingface_hub-0.23.5.tar.gz", hash = "sha256:67a9caba79b71235be3752852ca27da86bd54311d2424ca8afdb8dda056edf98"},
 ]
 
 [package.dependencies]
@@ -1450,13 +1450,13 @@ telegram = ["requests"]
 
 [[package]]
 name = "transformers"
-version = "4.42.3"
+version = "4.42.4"
 description = "State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow"
 optional = false
 python-versions = ">=3.8.0"
 files = [
-    {file = "transformers-4.42.3-py3-none-any.whl", hash = "sha256:a61a0df9609b7d69229d941b2fd857c841ba3043d6da503d0da1a4b133f65b92"},
-    {file = "transformers-4.42.3.tar.gz", hash = "sha256:7539873ff45809145265cbc94ea4619d2713c41ceaa277b692d8b0be3430f7eb"},
+    {file = "transformers-4.42.4-py3-none-any.whl", hash = "sha256:6d59061392d0f1da312af29c962df9017ff3c0108c681a56d1bc981004d16d24"},
+    {file = "transformers-4.42.4.tar.gz", hash = "sha256:f956e25e24df851f650cb2c158b6f4352dfae9d702f04c113ed24fc36ce7ae2d"},
 ]
 
 [package.dependencies]
@@ -1572,4 +1572,4 @@ dev = ["black (>=19.3b0)", "pytest (>=4.6.2)"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.11"
-content-hash = "e58b659457f8a7dc54ca3f5e7c247351d90f7e741ecdbbf1fa94f4a597da8844"
+content-hash = "fa010523fc1b8c68319f79064436f0a308de0871614c195e63c21389a9155bac"
diff --git a/load_tests/pyproject.toml b/load_tests/pyproject.toml
index 1a3b0be0..a90249ff 100644
--- a/load_tests/pyproject.toml
+++ b/load_tests/pyproject.toml
@@ -17,6 +17,7 @@ jinja2 = "^3.1.4"
 transformers = "^4.42.3"
 gputil = "^1.4.0"
 tabulate = "^0.9.0"
+matplotlib = "^3.9.1"
 
 
 [build-system]