Skip to content

Commit f861c4a

Browse files
committed
chore: clean up advanced tutorials and update test configuration for JAX
1 parent 4c94603 commit f861c4a

File tree

5 files changed

+5
-1608
lines changed

5 files changed

+5
-1608
lines changed

brainpy/dnn/tests/test_linear.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414
# ==============================================================================
15+
16+
import os
17+
18+
os.environ['JAX_TRACEBACK_FILTERING'] = 'off'
19+
1520
import jax.numpy as jnp
1621
from absl.testing import absltest
1722
from absl.testing import parameterized

docs/advanced_tutorials.rst

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -24,17 +24,6 @@ Interoperation
2424
tutorial_advanced/integrate_bp_convlstm_into_flax.ipynb
2525

2626

27-
Brain Dynamics Dedicated Operators
28-
----------------------------------
29-
30-
.. toctree::
31-
:maxdepth: 1
32-
33-
tutorial_advanced/operator_custom_with_numba.ipynb
34-
tutorial_advanced/operator_custom_with_taichi.ipynb
35-
tutorial_advanced/operator_custom_with_cupy.ipynb
36-
37-
3827
Developer Guides
3928
----------------
4029

docs/tutorial_advanced/operator_custom_with_cupy.ipynb

Lines changed: 0 additions & 322 deletions
This file was deleted.

0 commit comments

Comments
 (0)