Use relative path for symbol import

apache · Aug 10, 2018 · c9bb406 · c9bb406
1 parent c053262
commit c9bb406
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/mxnet/contrib/tensorrt.py b/python/mxnet/contrib/tensorrt.py
@@ -21,7 +21,7 @@
 import logging
 import os
 
-from mxnet.symbol import Symbol
+from .. import symbol as sym
 
 from ..base import _LIB, SymbolHandle, MXNetError
 from ..base import check_call
@@ -65,7 +65,7 @@ def get_optimized_symbol(executor):
     handle = SymbolHandle()
     try:
         check_call(_LIB.MXExecutorGetOptimizedSymbol(executor.handle, ctypes.byref(handle)))
-        result = Symbol(handle=handle)
+        result = sym.Symbol(handle=handle)
         return result
     except MXNetError:
         logging.error('Error while trying to fetch TRT optimized symbol for graph. Please ensure '