chore: Disable fx to test workflow

Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

py/torch_tensorrt/__init__.py

@@ -7,6 +7,11 @@
 import ctypes
 import torch
 
+try:
+    import tensorrt
+except:
+    pass
+
 from torch_tensorrt._version import __version__
 from torch_tensorrt._compile import *
 from torch_tensorrt._util import *
@@ -17,7 +22,7 @@
 from torch_tensorrt._Input import Input
 from torch_tensorrt._Device import Device
 
-from torch_tensorrt import fx
+#from torch_tensorrt import fx
 
 def _register_with_torch():
     trtorch_dir = os.path.dirname(__file__)

py/torch_tensorrt/_compile.py

@@ -5,9 +5,9 @@
 import torch
 import torch.fx
 from enum import Enum
-import torch_tensorrt.fx
-from torch_tensorrt.fx.lower import lower_to_trt
-from torch_tensorrt.fx.utils import LowerPrecision
+#import torch_tensorrt.fx
+#from torch_tensorrt.fx.lower import lower_to_trt
+#from torch_tensorrt.fx.utils import LowerPrecision
 
 class _IRType(Enum):
     """Enum to set the minimum required logging level to print a message to stdout