Skip to content

Commit 8d8cbfd

Browse files
committed
chore: linter fixes
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent c864096 commit 8d8cbfd

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

tests/cpp/test_compiled_modules.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,8 @@ TEST_P(CppAPITests, CompiledModuleIsClose) {
4141
}
4242

4343
for (size_t i = 0; i < trt_results.size(); i++) {
44-
ASSERT_TRUE(torch_tensorrt::tests::util::cosineSimEqual(jit_results[i], trt_results[i].reshape_as(jit_results[i]), 0.99));
44+
ASSERT_TRUE(
45+
torch_tensorrt::tests::util::cosineSimEqual(jit_results[i], trt_results[i].reshape_as(jit_results[i]), 0.99));
4546
}
4647
}
4748

tests/cpp/test_multiple_registered_engines.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,13 @@ TEST(CppAPITest, CanRunMultipleEngines) {
5656
trt2_results.push_back(trt2_results_ivalues.toTensor());
5757

5858
for (size_t i = 0; i < trt1_results.size(); i++) {
59-
ASSERT_TRUE(torch_tensorrt::tests::util::cosineSimEqual(jit1_results[i], trt1_results[i].reshape_as(jit1_results[i]), 0.99));
59+
ASSERT_TRUE(torch_tensorrt::tests::util::cosineSimEqual(
60+
jit1_results[i], trt1_results[i].reshape_as(jit1_results[i]), 0.99));
6061
}
6162

6263
for (size_t i = 0; i < trt2_results.size(); i++) {
63-
ASSERT_TRUE(torch_tensorrt::tests::util::cosineSimEqual(jit2_results[i], trt2_results[i].reshape_as(jit2_results[i]), 0.99));
64+
ASSERT_TRUE(torch_tensorrt::tests::util::cosineSimEqual(
65+
jit2_results[i], trt2_results[i].reshape_as(jit2_results[i]), 0.99));
6466
}
6567
}
6668
#endif

0 commit comments

Comments
 (0)