[math] depress taichi import logging by forcing using `import_taichi(…

…)` utility, move taichi random functions into another file
brainpy · Dec 9, 2023 · bffcd0a · bffcd0a
1 parent 4eb88c1
commit bffcd0a
Show file tree

Hide file tree

Showing 8 changed files with 1,008 additions and 1,111 deletions.
diff --git a/brainpy/_src/dependency_check.py b/brainpy/_src/dependency_check.py
@@ -17,7 +17,7 @@
 
 taichi_install_info = (f'We need taichi=={_minimal_taichi_version}. '
                        f'Currently you can install taichi=={_minimal_taichi_version} through:\n\n'
-                       '> pip install taichi==1.7.0 -U')
+                       '> pip install taichi==1.7.0')
 os.environ["TI_LOG_LEVEL"] = "error"
 
 

diff --git a/brainpy/_src/math/event/_csr_matvec_taichi.py b/brainpy/_src/math/event/_csr_matvec_taichi.py
@@ -5,14 +5,16 @@
 import jax
 import jax.numpy as jnp
 import numpy as np
-import taichi as ti
 from jax.interpreters import ad
 
+from brainpy._src.dependency_check import import_taichi
 from brainpy._src.math.interoperability import as_jax
 from brainpy._src.math.op_register import XLACustomOp
 from brainpy._src.math.sparse._csr_mv_taichi import csrmv_taichi as normal_csrmv_taichi
 from brainpy._src.math.sparse._utils import csr_to_coo
 
+ti = import_taichi()
+
 __all__ = [
   'csrmv_taichi'
 ]