[Refactor] Avoid exposing real and integer types API

python/taichi/lang/__init__.py

@@ -57,8 +57,7 @@
 from taichi.snode.fields_builder import FieldsBuilder
 from taichi.tools.util import set_gdb_trigger, warning
 from taichi.types.annotations import any_arr, ext_arr, template
-from taichi.types.primitive_types import (f16, f32, f64, i32, i64,
-                                          integer_types, u32, u64)
+from taichi.types.primitive_types import f16, f32, f64, i32, i64, u32, u64
 
 from taichi import _logging
 

python/taichi/lang/struct.py

@@ -8,7 +8,7 @@
 from taichi.lang.matrix import Matrix
 from taichi.lang.util import (cook_dtype, in_python_scope, is_taichi_class,
                               python_scope, taichi_scope)
-from taichi.types import CompoundType, integer_types
+from taichi.types import CompoundType, primitive_types
 
 
 class Struct(TaichiOperations):
@@ -494,7 +494,9 @@ def cast(self, struct):
             else:
                 if in_python_scope():
                     v = struct.entries[k]
-                    entries[k] = int(v) if dtype in integer_types else float(v)
+                    entries[k] = int(
+                        v
+                    ) if dtype in primitive_types.integer_types else float(v)
                 else:
                     entries[k] = ops.cast(struct.entries[k], dtype)
         return Struct(entries)

python/taichi/types/primitive_types.py

@@ -90,8 +90,6 @@
     'u32',
     'uint64',
     'u64',
-    'real_types',
-    'integer_types',
     'is_signed',
     'is_integral',
 ]