mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
Fix lit flake8 errors
This commit is contained in:
@@ -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(','))
|
||||
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user