[Refactor] Remove unnecessary exposure related to matrix and mesh

taichi-dev · Jan 26, 2022 · 0056369 · 0056369
1 parent 3305f66
commit 0056369
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/python/taichi/lang/__init__.py b/python/taichi/lang/__init__.py
@@ -33,11 +33,10 @@
                               static, static_assert, static_print, stop_grad,
                               subscript, ti_assert, ti_float, ti_format,
                               ti_int, ti_print, zero)
-from taichi.lang.kernel_arguments import SparseMatrixProxy
 from taichi.lang.kernel_impl import (KernelArgError, KernelDefError,
                                      data_oriented, func, kernel, pyfunc)
-from taichi.lang.matrix import Matrix, MatrixField, Vector
-from taichi.lang.mesh import Mesh, MeshElementFieldProxy, TetMesh, TriMesh
+from taichi.lang.matrix import *
+from taichi.lang.mesh import *
 from taichi.lang.ops import *  # pylint: disable=W0622
 from taichi.lang.quant_impl import quant
 from taichi.lang.runtime_ops import async_flush, sync

diff --git a/python/taichi/lang/matrix.py b/python/taichi/lang/matrix.py
@@ -1429,3 +1429,6 @@ def fill_by_kernel(self, val):
 
     def __repr__(self):
         return f'<{self.n} {self.layout} ti.Vector.ndarray>'
+
+
+__all__ = ["Matrix", "Vector"]
diff --git a/python/taichi/lang/mesh.py b/python/taichi/lang/mesh.py
@@ -528,3 +528,6 @@ def subscript(self, *indices):
         entry_expr.type_check()
         return MeshElementFieldProxy(self.mesh, self.to_element_type,
                                      entry_expr)
+
+
+__all__ = ["Mesh", "TetMesh", "TriMesh"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -1429,3 +1429,6 @@ def fill_by_kernel(self, val):

		def __repr__(self):
		return f'<{self.n} {self.layout} ti.Vector.ndarray>'


		__all__ = ["Matrix", "Vector"]