Skip to content

Commit febb438

Browse files
committed
rename files
1 parent c6f6939 commit febb438

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def run(self):
7474
version=VersionInfo.version,
7575
description='Tensorflow to ONNX converter',
7676
setup_requires=['pytest-runner'],
77-
tests_require=['graphviz', 'parameterized', 'pytest', 'pytest-cov', 'pyyaml'],
77+
tests_require=['graphviz', 'parameterized', 'pytest', 'pytest-cov', 'pyyaml', 'fire'],
7878
cmdclass=cmdclass,
7979
packages=find_packages(),
8080

tests/test_benchmarks.py renamed to tests/test_profile.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@
1111
# pylint: disable=invalid-name
1212
# pylint: enable=invalid-name
1313

14-
class BenchmarksTests(Tf2OnnxBackendTestBase):
14+
class ProfileTests(Tf2OnnxBackendTestBase):
1515

16-
folder = os.path.join(os.path.dirname(__file__), '..', 'benchmarks')
16+
folder = os.path.join(os.path.dirname(__file__), '..', 'tools')
1717

1818
@check_opset_after_tf_version("2.0", 12, "might need Scan")
1919
def test_profile_conversion_time(self):
20-
filename = os.path.join(BenchmarksTests.folder, 'profile_conversion_time.py')
20+
filename = os.path.join(ProfileTests.folder, 'profile_conversion_time.py')
2121
proc = subprocess.Popen(
2222
["python", filename], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
2323
try:

0 commit comments

Comments
 (0)