Update dependency_check.py

brainpy · May 8, 2024 · e9c21a4 · e9c21a4
1 parent 67be1de
commit e9c21a4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/brainpy/_src/dependency_check.py b/brainpy/_src/dependency_check.py
@@ -57,9 +57,11 @@ def import_taichi(error_if_not_found=True):
 
   if taichi is None:
     return None
-  if taichi.__version__ != _minimal_taichi_version:
-    raise RuntimeError(taichi_install_info)
-  return taichi
+  if taichi.__version__[0] >= _minimal_taichi_version[0] and taichi.__version__[1] >= _minimal_taichi_version[1] and \
+      taichi.__version__[2] >= _minimal_taichi_version[2]:
+    return taichi
+  else:
+    raise ModuleNotFoundError(taichi_install_info)
 
 
 def raise_taichi_not_found(*args, **kwargs):
@@ -182,4 +184,4 @@ def import_brainpylib_gpu_ops():
       raise ImportError('Please install GPU version of brainpylib. \n'
                         'See https://brainpy.readthedocs.io for installation instructions.')
 
-  return brainpylib_gpu_ops
+  return brainpylib_gpu_ops