Fix lit flake8 errors

This commit is contained in:
anush elangovan
2022-06-02 18:56:55 +00:00
parent 395adc2921
commit c6ea400178
2 changed files with 1 additions and 47 deletions

View File

@@ -1,46 +0,0 @@
import os
import sys
import lit.formats
import lit.util
import lit.llvm
# Configuration file for the 'lit' test runner.
lit.llvm.initialize(lit_config, config)
# name: The name of this test suite.
config.name = 'TFLITEHUB'
config.test_format = lit.formats.ShTest()
# suffixes: A list of file extensions to treat as test files.
config.suffixes = ['.py']
# test_source_root: The root path where tests are located.
config.test_source_root = os.path.dirname(__file__)
#config.use_default_substitutions()
config.excludes = [
'coco_test_data.py',
'imagenet_test_data.py',
'lit.cfg.py',
'lit.site.cfg.py',
'manual_test.py',
'squad_test_data.py',
'test_util.py',
]
config.substitutions.extend([
('%PYTHON', sys.executable),
])
config.environment['PYTHONPATH'] = ":".join(sys.path)
project_root = os.path.dirname(os.path.dirname(__file__))
# Enable features based on -D FEATURES=hugetest,vulkan
# syntax.
features_param = lit_config.params.get('FEATURES')
if features_param:
config.available_features.update(features_param.split(','))

View File

@@ -13,7 +13,7 @@ class SsdMobilenetV2Test(test_util.TFLiteModelTest):
super(SsdMobilenetV2Test, self).__init__(model_path, *args, **kwargs)
def compare_results(self, iree_results, tflite_results, details):
super(SsdMobilenetV1Test, self).compare_results(iree_results, tflite_results, details)
super(SsdMobilenetV2Test, self).compare_results(iree_results, tflite_results, details)
for i in range(len(iree_results)):
# Dequantize outputs.
zero_point = details[i]['quantization_parameters']['zero_points'][0]