Adjust parameter in test_GaussianProb_opt.py and Remove seaborn requirement

Author: Routhleck

brainpy/_src/connect/base.py

@@ -9,8 +9,6 @@
 from brainpy import tools, math as bm
 from brainpy.errors import ConnectorError
 
-import matplotlib.pyplot as plt
-import seaborn as sns
 import textwrap
 
 __all__ = [
@@ -729,6 +727,12 @@ def coo2csc(coo, post_num, data=None):
 
 
 def visualizeMat(mat, description):
+  try:
+    import seaborn as sns
+    import matplotlib.pyplot as plt
+  except (ModuleNotFoundError, ImportError):
+    print('Please install seaborn and matplotlib for this function')
+    return
   sns.heatmap(mat, cmap='viridis')
   warpped_title = textwrap.fill(description, width=60)
   plt.title(warpped_title)

brainpy/_src/connect/tests/test_GaussianProb_opt.py

@@ -10,7 +10,7 @@
 
 
 def test_gaussian_prob1():
-    conn = bp.connect.GaussianProb(sigma=1., include_self=False, seed=123)(pre_size=2000)
+    conn = bp.connect.GaussianProb(sigma=1., include_self=False, seed=123)(pre_size=100)
 
     mat = conn.build_mat(isOptimized=True)
     time0 = time()
@@ -27,7 +27,7 @@ def test_gaussian_prob1():
 
 
 def test_gaussian_prob2():
-    conn = bp.connect.GaussianProb(sigma=4, seed=123)(pre_size=(100, 100))
+    conn = bp.connect.GaussianProb(sigma=4, seed=123)(pre_size=(10, 10))
     mat = conn.build_mat(isOptimized=True)
     time0 = time()
     mat1 = conn.build_mat(isOptimized=True)
@@ -43,7 +43,7 @@ def test_gaussian_prob2():
 
 
 def test_gaussian_prob3():
-    conn = bp.connect.GaussianProb(sigma=4, periodic_boundary=True, seed=123)(pre_size=(200, 200))
+    conn = bp.connect.GaussianProb(sigma=4, periodic_boundary=True, seed=123)(pre_size=(10, 10))
     mat = conn.build_mat(isOptimized=True)
     time0 = time()
     mat1 = conn.build_mat(isOptimized=True)
@@ -59,7 +59,7 @@ def test_gaussian_prob3():
 
 
 def test_gaussian_prob4():
-    conn = bp.connect.GaussianProb(sigma=4, periodic_boundary=True, seed=123)(pre_size=(25, 25, 25))
+    conn = bp.connect.GaussianProb(sigma=4, periodic_boundary=True, seed=123)(pre_size=(10, 10, 10))
     mat = conn.build_mat(isOptimized=True)
     time0 = time()
     mat1 = conn.build_mat(isOptimized=True)

requirements-dev.txt

@@ -8,7 +8,6 @@ scipy>=1.1.0
 brainpylib
 h5py
 pathos
-seaborn
 
 # test requirements
 pytest