[hotfix] pytest error of act_act example (#12093)

* [hotfix] pytest error of act_act example

* remove needless import
This commit is contained in:
Bonhun Koo 2022-07-11 12:56:32 +09:00 committed by GitHub
parent bbf54e01e7
commit b371e74735
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions

View File

@ -1,3 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:e6eb24c47d38ae992eef207a5c4e873db496a63eed42ac5469df4f2afc756f48
size 16835
oid sha256:f44cbc66f71d53879ead60e9db65a80c3d9b572323aca87265567a01ca77426d
size 16673

View File

@ -15,14 +15,14 @@ from .utils.check_graph import check_model
from .utils.config import merge_configs
TEST_MODELS = [
('resnet_example', 'pytorch')
('resnet_example', 'pytorch', 'Ranger', 'ANY'),
('resnet_example', 'pytorch', 'DefaultQuantization', 'CPU'),
('resnet_example', 'pytorch', 'RangeSupervision', 'ANY'),
('act_act_example', 'pytorch', 'DefaultQuantization', 'GNA')
]
@pytest.fixture(scope='module', params=TEST_MODELS,
ids=['{}_{}_{}'.format(*m) for m in TEST_MODELS])
ids=['_'.join(m) for m in TEST_MODELS])
def _params(request):
return request.param
@ -32,7 +32,7 @@ def test_graph_with_stats(_params, tmp_path, models):
algorithm_config = Dict({
'algorithms': [{
'name': 'algo_name',
'name': algo_name,
'params': {
'target_device': target_device,
'preset': 'performance',