Skip to content

Commit 86ce9ca

Browse files
authored
Remove: failing test as it was only extracting a value from the response body (#306)
1 parent 00ccfac commit 86ce9ca

File tree

1 file changed

+1
-20
lines changed

1 file changed

+1
-20
lines changed

tests/sparsezoo/deployment_package/utils/test_extractors.py

+1-20
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,7 @@
1717
import pytest
1818

1919
from sparsezoo import Model
20-
from sparsezoo.deployment_package.utils.extractors import (
21-
_batch_size_one_latency,
22-
_throughput,
23-
)
20+
from sparsezoo.deployment_package.utils.extractors import _batch_size_one_latency
2421

2522

2623
_VNNI_STUB = (
@@ -63,22 +60,6 @@ def model():
6360
yield Model(source=stub)
6461

6562

66-
@pytest.mark.parametrize(
67-
"num_cores,batch_size,device_info,expected",
68-
[
69-
(24, 64, "c6i.12xlarge", 1948.45),
70-
],
71-
)
72-
def test_throughput_extractor(model, num_cores, batch_size, device_info, expected):
73-
actual_throughput = _throughput(
74-
model=model,
75-
num_cores=num_cores,
76-
batch_size=batch_size,
77-
device_info=device_info,
78-
)
79-
assert actual_throughput == expected
80-
81-
8263
@patch("sparsezoo.deployment_package_module.utils.extractors._throughput")
8364
def test_latency_extractor(throughput_patch, model):
8465
throughput_patch.return_value = 0

0 commit comments

Comments
 (0)