Update python/taichi/lang/matrix_ops_utils.py

Co-authored-by: Zhanlue Yang <[email protected]>
taichi-dev · Nov 3, 2022 · 53bddaa · 53bddaa
1 parent 13041f5
commit 53bddaa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/taichi/lang/matrix_ops_utils.py b/python/taichi/lang/matrix_ops_utils.py
@@ -75,6 +75,8 @@ def assert_tensor(m, msg='not tensor type: {}'):
     raise TaichiCompilationError(msg.format(type(m)))
 
 
+# TODO(zhanlue): rearrange to more generic checker functions
+# for example: "assert_is_instance(args, indices=[], instances=[], logic='or')"
 def assert_vector(v, msg='not a vector: {}'):
     if (isinstance(v, Expr) or isinstance(v, Matrix)) and len(
             v.get_shape()) == 1: