Auto Format

taichi-dev · Dec 24, 2021 · ab3837a · ab3837a
1 parent d091224
commit ab3837a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/python/test_customized_grad.py b/tests/python/test_customized_grad.py
@@ -187,6 +187,7 @@ def func(mul: ti.f32):
             ti.atomic_add(total[None], x[i] * mul)
 
     with pytest.raises(RuntimeError):
+
         @ti.ad.grad_for(func)
         def backward(mul):
             func.grad(mul)
@@ -216,6 +217,7 @@ def foward(mul):
         func(mul)
 
     with pytest.raises(RuntimeError):
+
         @ti.ad.grad_for(func)
         def backward(mul):
             func.grad(mul)

diff --git a/tests/python/test_offset.py b/tests/python/test_offset.py
@@ -132,5 +132,9 @@ def test_offset_must_throw_vector():
 @ti.test()
 def test_offset_must_throw_matrix():
     with pytest.raises(AssertionError):
-        c = ti.Matrix.field(3, 3, dtype=ti.i32, shape=(32, 16, 8), offset=(32, 16))
+        c = ti.Matrix.field(3,
+                            3,
+                            dtype=ti.i32,
+                            shape=(32, 16, 8),
+                            offset=(32, 16))
         d = ti.Matrix.field(3, 3, dtype=ti.i32, shape=None, offset=(32, 16))