fix: Missing log severity

narendasan · narendasan · commit 6a4daef6ad3b · 2021-11-10T16:57:44.000-08:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
@@ -143,6 +143,7 @@ def convert_method_to_trt_engine(module: Any,
         ts_mod = module
         if isinstance(module, torch.nn.Module):
             logging.log(
+                logging.Level.Info,
                 "Module was provided as a torch.nn.Module, trying to script the module with torch.jit.script. In the event of a failure please preconvert your module to TorchScript"
             )
             ts_mod = torch.jit.script(module)

Original file line number	Diff line number	Diff line change
`@@ -143,6 +143,7 @@ def convert_method_to_trt_engine(module: Any,`
`143`	`143`	`ts_mod = module`
`144`	`144`	`if isinstance(module, torch.nn.Module):`
`145`	`145`	`logging.log(`
	`146`	`+ logging.Level.Info,`
`146`	`147`	`"Module was provided as a torch.nn.Module, trying to script the module with torch.jit.script. In the event of a failure please preconvert your module to TorchScript"`
`147`	`148`	`)`
`148`	`149`	`ts_mod = torch.jit.script(module)`