Fix ModuleNotFoundError in ptq (#2814)

HolyWu · web-flow · commit 07c5b07b5ca8 · 2024-05-06T10:50:07.000-06:00
diff --git a/py/torch_tensorrt/ts/ptq.py b/py/torch_tensorrt/ts/ptq.py
@@ -11,7 +11,7 @@
 
 import torch
 from torch_tensorrt import _C
-from py.torch_tensorrt.ts.logging import Level, log
+from torch_tensorrt.ts.logging import Level, log
 
 
 class CalibrationAlgo(Enum):
