reformatted by black

apache · Aug 11, 2022 · 66a74e3 · 66a74e3
1 parent de4fd9c
commit 66a74e3
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py
@@ -126,8 +126,15 @@ def load_model(model_name):
 
 
 def verify_model(
-    model_name, input_data=None, custom_convert_map=None, rtol=1e-5, atol=1e-5, expected_ops=None,
-    kind="graph", check_correctness=True, cpu_only=False
+    model_name,
+    input_data=None,
+    custom_convert_map=None,
+    rtol=1e-5,
+    atol=1e-5,
+    expected_ops=None,
+    kind="graph",
+    check_correctness=True,
+    cpu_only=False,
 ):
     """Assert that the output of a compiled model matches with that of its
     baseline."""