From da3fd35b4d16d84c8df474dea9f029faac8449c8 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 19 Jan 2023 10:06:12 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 python/taichi/aot/utils.py                     | 4 +++-
 tests/cpp/aot/python_scripts/graph_aot_test.py | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/python/taichi/aot/utils.py b/python/taichi/aot/utils.py
index 28d6982be3716c..c715347cb23851 100644
--- a/python/taichi/aot/utils.py
+++ b/python/taichi/aot/utils.py
@@ -13,7 +13,9 @@
 
 def check_type_match(lhs, rhs):
     if isinstance(lhs, MatrixType) and isinstance(rhs, MatrixType):
-        return lhs.n == rhs.n and lhs.m == rhs.m and (lhs.dtype == rhs.dtype or lhs.dtype is None or rhs.dtype is None)
+        return lhs.n == rhs.n and lhs.m == rhs.m and (lhs.dtype == rhs.dtype
+                                                      or lhs.dtype is None
+                                                      or rhs.dtype is None)
     elif isinstance(lhs, MatrixType) or isinstance(rhs, MatrixType):
         return False
     else:
diff --git a/tests/cpp/aot/python_scripts/graph_aot_test.py b/tests/cpp/aot/python_scripts/graph_aot_test.py
index 284ecf190a5ab7..ae9afd0380e52a 100644
--- a/tests/cpp/aot/python_scripts/graph_aot_test.py
+++ b/tests/cpp/aot/python_scripts/graph_aot_test.py
@@ -16,7 +16,9 @@ def run0(base: int, arr: ti.types.ndarray(ndim=1, dtype=ti.i32)):
             arr[i] += base + i
 
     @ti.kernel
-    def run1(base: int, arr: ti.types.ndarray(ndim=1, dtype=ti.types.vector(1, ti.i32))):
+    def run1(base: int, arr: ti.types.ndarray(ndim=1,
+                                              dtype=ti.types.vector(1,
+                                                                    ti.i32))):
         for i in arr:
             arr[i] += base + i