chore: Address BERT TS failure

peri044 · peri044 · commit 6745b761105e · 2023-07-18T18:28:38.000-07:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/tests/py/ts/models/test_models.py b/tests/py/ts/models/test_models.py
@@ -110,7 +110,6 @@ def test_bert_base_uncased(self):
             },
             "enabled_precisions": {torch.float},
             "truncate_long_and_double": True,
-            "ir": "ts",
         }
         with torchtrt.logging.errors():
             trt_mod = torchtrt.ts.compile(self.model, **compile_spec)

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,6 @@ def test_bert_base_uncased(self):`
`110`	`110`	`},`
`111`	`111`	`"enabled_precisions": {torch.float},`
`112`	`112`	`"truncate_long_and_double": True,`
`113`		`- "ir": "ts",`
`114`	`113`	`}`
`115`	`114`	`with torchtrt.logging.errors():`
`116`	`115`	`trt_mod = torchtrt.ts.compile(self.model, **compile_spec)`