Skip to content

Commit

Permalink
Move the pylint message and fix comment length
Browse files Browse the repository at this point in the history
  • Loading branch information
aaroey committed Mar 7, 2018
1 parent 6dd28ea commit f7b1d23
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tensorflow/contrib/tensorrt/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,16 +18,16 @@
from __future__ import division
from __future__ import print_function

# pylint: disable=unused-import,wildcard-import
# pylint: disable=unused-import,wildcard-import,g-import-not-at-top
try:
from tensorflow.contrib.tensorrt.python import * # pylint: disable=import-not-at-top
from tensorflow.contrib.tensorrt.python import *
except Exception as e:
no_trt_message = (
'**** Failed to initialize TensorRT. This is either because the TensorRT'
' installation path is not in LD_LIBRARY_PATH, or because you do not have it'
' installed. If not installed, please go to'
' installation path is not in LD_LIBRARY_PATH, or because you do not have'
' it installed. If not installed, please go to'
' https://developer.nvidia.com/tensorrt to download and install'
' TensorRT ****')
print(no_trt_message)
raise e
# pylint: enable=unused-import,wildcard-import
# pylint: enable=unused-import,wildcard-import,g-import-not-at-top

0 comments on commit f7b1d23

Please sign in to comment.