[Refactor] Fix typo and type in docstring, and format too long string

python/taichi/lang/kernel_impl.py

@@ -286,14 +286,14 @@ def extract_arg(arg, anno):
             shape = tuple(arg.shape)
             if len(shape) < element_dim:
                 raise ValueError(
-                    f"Invalid argument into ti.any_arr() - required element_dim={element_dim}, but the argument has only {len(shape)} dimensions"
-                )
+                    f"Invalid argument into ti.any_arr() - required element_dim={element_dim}, "
+                    f"but the argument has only {len(shape)} dimensions")
             element_shape = (
             ) if element_dim == 0 else shape[:
                                              element_dim] if layout == Layout.SOA else shape[
                                                  -element_dim:]
             return to_taichi_type(arg.dtype), len(shape), element_shape, layout
-        return (type(arg).__name__, )
+        return type(arg).__name__,
 
     def extract(self, args):
         extracted = []
@@ -451,8 +451,10 @@ def taichi_ast_generator():
             _taichi_skip_traceback = 1
             if self.runtime.inside_kernel:
                 raise TaichiSyntaxError(
-                    "Kernels cannot call other kernels. I.e., nested kernels are not allowed. Please check if you have direct/indirect invocation of kernels within kernels. Note that some methods provided by the Taichi standard library may invoke kernels, and please move their invocations to Python-scope."
-                )
+                    "Kernels cannot call other kernels. I.e., nested kernels are not allowed. "
+                    "Please check if you have direct/indirect invocation of kernels within kernels. "
+                    "Note that some methods provided by the Taichi standard library may invoke kernels, "
+                    "and please move their invocations to Python-scope.")
             self.runtime.inside_kernel = True
             self.runtime.current_kernel = self
             try:

python/taichi/lang/matrix.py

@@ -26,9 +26,9 @@ class Matrix(TaichiOperations):
     """The matrix class.
 
     Args:
-        n (int): the first dimension of a matrix.
+        n (Union[int, list, tuple], np.ndarray): the first dimension of a matrix.
         m (int): the second dimension of a matrix.
-        dt (DataType): the elmement data type.
+        dt (DataType): the element data type.
         keep_raw (Bool, optional): Keep the contents in `n` as is.
     """
     is_taichi_class = True
@@ -145,8 +145,8 @@ def __init__(self,
                 self.m = m
             else:
                 raise ValueError(
-                    "Declaring matrix fields using `ti.Matrix(n, m, dt, shape)` is no longer supported. Use `ti.Matrix.field(n, m, dtype, shape)` instead."
-                )
+                    "Declaring matrix fields using `ti.Matrix(n, m, dt, shape)` is no longer supported. "
+                    "Use `ti.Matrix.field(n, m, dtype, shape)` instead.")
 
         if self.n * self.m > 32 and not suppress_warning:
             warning(
@@ -611,7 +611,7 @@ def diag(dim, val):
 
         Args:
             dim (int): the dimension of a square matrix.
-            val (TypeVar): the diagonal elment value.
+            val (TypeVar): the diagonal element value.
 
         Returns:
             The constructed diagonal square matrix.
@@ -678,7 +678,7 @@ def max(self):
 
     @kern_mod.pyfunc
     def min(self):
-        """Return the minumum element value."""
+        """Return the minimum element value."""
         return ops_mod.ti_min(*self.entries)
 
     def any(self):
@@ -994,7 +994,7 @@ def _Vector_ndarray(cls, n, dtype, shape, layout=Layout.AOS):
 
     @staticmethod
     def rows(rows):
-        """Construct a Matrix instance by concactinating Vectors/lists row by row.
+        """Construct a Matrix instance by concatenating Vectors/lists row by row.
 
         Args:
             rows (List): A list of Vector (1-D Matrix) or a list of list.
@@ -1027,7 +1027,7 @@ def rows(rows):
 
     @staticmethod
     def cols(cols):
-        """Construct a Matrix instance by concactinating Vectors/lists column by column.
+        """Construct a Matrix instance by concatenating Vectors/lists column by column.
 
         Args:
             cols (List): A list of Vector (1-D Matrix) or a list of list.
@@ -1148,7 +1148,7 @@ def Vector(n, dt=None, **kwargs):
     """Construct a `Vector` instance i.e. 1-D Matrix.
 
     Args:
-        n (int): The desired number of entries of the Vector.
+        n (Union[int, list, tuple], np.ndarray): The desired number of entries of the Vector.
         dt (DataType, optional): The desired data type of the Vector.
 
     Returns:

python/taichi/misc/util.py

@@ -75,7 +75,7 @@ def core_vec(*args):
         assert False, type(args[0])
 
 
-class Tee():
+class Tee:
     def __init__(self, name):
         with open(name, 'w') as self.file:
             self.stdout = sys.stdout
@@ -256,6 +256,7 @@ def print_async_stats(include_kernel_profiler=False):
     'core_vec',
     'core_veci',
     'deprecated',
+    'warning',
     'dump_dot',
     'dot_to_pdf',
     'obsolete',