Skip to content

Commit 7d67da2

Browse files
committed
refactor: Apply linting
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent d2cc1e9 commit 7d67da2

File tree

1 file changed

+13
-30
lines changed

1 file changed

+13
-30
lines changed

py/setup.py

Lines changed: 13 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ def gen_version_file():
106106
print("creating version file")
107107
f.write("__version__ = \"" + __version__ + '\"')
108108

109+
109110
def copy_libtrtorch(multilinux=False):
110111
if not os.path.exists(dir_path + '/trtorch/lib'):
111112
os.makedirs(dir_path + '/trtorch/lib')
@@ -171,15 +172,8 @@ def run(self):
171172
class CleanCommand(Command):
172173
"""Custom clean command to tidy up the project root."""
173174
PY_CLEAN_FILES = [
174-
'./build',
175-
'./dist',
176-
'./trtorch/__pycache__',
177-
'./trtorch/lib',
178-
'./trtorch/include',
179-
'./trtorch/bin',
180-
'./*.pyc',
181-
'./*.tgz',
182-
'./*.egg-info'
175+
'./build', './dist', './trtorch/__pycache__', './trtorch/lib', './trtorch/include', './trtorch/bin', './*.pyc',
176+
'./*.tgz', './*.egg-info'
183177
]
184178
description = "Command to tidy up the project root"
185179
user_options = []
@@ -265,27 +259,16 @@ def run(self):
265259
python_requires='>=3.6',
266260
include_package_data=True,
267261
package_data={
268-
'trtorch': ['lib/*',
269-
'include/trtorch/*.h',
270-
'include/trtorch/core/*.h',
271-
'include/trtorch/core/conversion/*.h',
272-
'include/trtorch/core/conversion/conversionctx/*.h',
273-
'include/trtorch/core/conversion/converters/*.h',
274-
'include/trtorch/core/conversion/evaluators/*.h',
275-
'include/trtorch/core/conversion/tensorcontainer/*.h',
276-
'include/trtorch/core/conversion/var/*.h',
277-
'include/trtorch/core/ir/*.h',
278-
'include/trtorch/core/lowering/*.h',
279-
'include/trtorch/core/lowering/passes/*.h',
280-
'include/trtorch/core/partitioning/*.h',
281-
'include/trtorch/core/plugins/*.h',
282-
'include/trtorch/core/plugins/impl/*.h',
283-
'include/trtorch/core/runtime/*.h',
284-
'include/trtorch/core/util/*.h',
285-
'include/trtorch/core/util/logging/*.h',
286-
'bin/*',
287-
'BUILD',
288-
'WORKSPACE'],
262+
'trtorch': [
263+
'lib/*', 'include/trtorch/*.h', 'include/trtorch/core/*.h', 'include/trtorch/core/conversion/*.h',
264+
'include/trtorch/core/conversion/conversionctx/*.h', 'include/trtorch/core/conversion/converters/*.h',
265+
'include/trtorch/core/conversion/evaluators/*.h', 'include/trtorch/core/conversion/tensorcontainer/*.h',
266+
'include/trtorch/core/conversion/var/*.h', 'include/trtorch/core/ir/*.h',
267+
'include/trtorch/core/lowering/*.h', 'include/trtorch/core/lowering/passes/*.h',
268+
'include/trtorch/core/partitioning/*.h', 'include/trtorch/core/plugins/*.h',
269+
'include/trtorch/core/plugins/impl/*.h', 'include/trtorch/core/runtime/*.h',
270+
'include/trtorch/core/util/*.h', 'include/trtorch/core/util/logging/*.h', 'bin/*', 'BUILD', 'WORKSPACE'
271+
],
289272
},
290273
exclude_package_data={
291274
'': ['*.cpp'],

0 commit comments

Comments
 (0)