chore: minor fix

peri044 · peri044 · commit 8b7761701f7a · 2023-07-19T14:44:06.000-07:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
@@ -69,7 +69,7 @@ def _get_target_ir(module_type: _ModuleType, ir: str) -> _IRType:
             elif module_is_tsable:
                 logging.log(
                     logging.Level.Warning,
-                    "Input graph is a Torchscript module but the ir provided is default (dynamo). Please set ir=torchscript to suppress the warning. Compiling the module with ir=ts",
+                    "Input graph is a Torchscript module but the ir provided is default (dynamo). Please set ir=torchscript to suppress the warning. Compiling the module with ir=torchscript",
                 )
                 return _IRType.ts
             else:

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ def _get_target_ir(module_type: _ModuleType, ir: str) -> _IRType:`
`69`	`69`	`elif module_is_tsable:`
`70`	`70`	`logging.log(`
`71`	`71`	`logging.Level.Warning,`
`72`		`- "Input graph is a Torchscript module but the ir provided is default (dynamo). Please set ir=torchscript to suppress the warning. Compiling the module with ir=ts",`
	`72`	`+ "Input graph is a Torchscript module but the ir provided is default (dynamo). Please set ir=torchscript to suppress the warning. Compiling the module with ir=torchscript",`
`73`	`73`	`)`
`74`	`74`	`return _IRType.ts`
`75`	`75`	`else:`