huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-23 16:02:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fix to allow report for a full failed test

Browse Source
This commit is contained in:
Hugo Larcher 2024-07-18 10:03:05 +02:00
parent cf70a3036b
commit 91a8972e18
No known key found for this signature in database
GPG Key ID: 3DAF63124699CA2B
3 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
load_tests/parse_load_test.py
View File

@ -20,12 +20,13 @@ def get_nested(obj, path, default=None):
for key in path.split("."): for key in path.split("."):
if obj is None: if obj is None:
return default return default
if re.match(r"\d+", key): if isinstance(obj, list):
try:
key = int(key) key = int(key)
if key >= len(obj): except ValueError:
return default return default
obj = obj[key] obj = obj[key]
else: elif isinstance(obj, dict):
obj = obj.get(key, default) obj = obj.get(key, default)
return obj return obj

14
load_tests/poetry.lock generated
View File

@ -353,13 +353,13 @@ files = [
[[package]] [[package]]
name = "huggingface-hub" 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" description = "Client library to download and publish models, datasets and other repos on the huggingface.co hub"
optional = false optional = false
python-versions = ">=3.8.0" python-versions = ">=3.8.0"
files = [ files = [
{file = "huggingface_hub-0.23.4-py3-none-any.whl", hash = "sha256:3a0b957aa87150addf0cc7bd71b4d954b78e749850e1e7fb29ebbd2db64ca037"}, {file = "huggingface_hub-0.23.5-py3-none-any.whl", hash = "sha256:d7a7d337615e11a45cc14a0ce5a605db6b038dc24af42866f731684825226e90"},
{file = "huggingface_hub-0.23.4.tar.gz", hash = "sha256:35d99016433900e44ae7efe1c209164a5a81dbbcd53a52f99c281dcd7ce22431"}, {file = "huggingface_hub-0.23.5.tar.gz", hash = "sha256:67a9caba79b71235be3752852ca27da86bd54311d2424ca8afdb8dda056edf98"},
] ]
[package.dependencies] [package.dependencies]
@ -1450,13 +1450,13 @@ telegram = ["requests"]
[[package]] [[package]]
name = "transformers" name = "transformers"
version = "4.42.3" version = "4.42.4"
description = "State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow" description = "State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow"
optional = false optional = false
python-versions = ">=3.8.0" python-versions = ">=3.8.0"
files = [ files = [
{file = "transformers-4.42.3-py3-none-any.whl", hash = "sha256:a61a0df9609b7d69229d941b2fd857c841ba3043d6da503d0da1a4b133f65b92"}, {file = "transformers-4.42.4-py3-none-any.whl", hash = "sha256:6d59061392d0f1da312af29c962df9017ff3c0108c681a56d1bc981004d16d24"},
{file = "transformers-4.42.3.tar.gz", hash = "sha256:7539873ff45809145265cbc94ea4619d2713c41ceaa277b692d8b0be3430f7eb"}, {file = "transformers-4.42.4.tar.gz", hash = "sha256:f956e25e24df851f650cb2c158b6f4352dfae9d702f04c113ed24fc36ce7ae2d"},
] ]
[package.dependencies] [package.dependencies]
@ -1572,4 +1572,4 @@ dev = ["black (>=19.3b0)", "pytest (>=4.6.2)"]
[metadata] [metadata]
lock-version = "2.0" lock-version = "2.0"
python-versions = "^3.11" python-versions = "^3.11"
content-hash = "e58b659457f8a7dc54ca3f5e7c247351d90f7e741ecdbbf1fa94f4a597da8844" content-hash = "fa010523fc1b8c68319f79064436f0a308de0871614c195e63c21389a9155bac"

1
load_tests/pyproject.toml
View File

@ -17,6 +17,7 @@ jinja2 = "^3.1.4"
transformers = "^4.42.3" transformers = "^4.42.3"
gputil = "^1.4.0" gputil = "^1.4.0"
tabulate = "^0.9.0" tabulate = "^0.9.0"
matplotlib = "^3.9.1"
[build-system] [build-system]