Enable brainpy object as pytree so that it can be applied with jax.jit etc. directly
#968
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Continuous Integration with Models | |
| on: | |
| push: | |
| branches: | |
| - '**' # matches every branch | |
| pull_request: | |
| branches: | |
| - '**' # matches every branch | |
| # | |
| #on: | |
| # push: | |
| # branches: [ master ] | |
| # pull_request: | |
| # branches: [ master ] | |
| jobs: | |
| test_linux: | |
| runs-on: ubuntu-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| python-version: [ "3.9", "3.10", "3.11"] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Install dependencies | |
| run: | | |
| if [ -f requirements-dev.txt ]; then pip install -r requirements-dev.txt; fi | |
| pip uninstall brainpy -y | |
| python setup.py install | |
| - name: Test with pytest | |
| run: | | |
| pytest tests/ | |
| # test_linux_py37: | |
| # runs-on: ubuntu-latest | |
| # if: github.event.pull_request.merged == true | |
| # strategy: | |
| # fail-fast: false | |
| # matrix: | |
| # python-version: ["3.7"] | |
| # | |
| # steps: | |
| # - uses: actions/checkout@v4 | |
| # - name: Set up Python ${{ matrix.python-version }} | |
| # uses: actions/setup-python@v5 | |
| # with: | |
| # python-version: ${{ matrix.python-version }} | |
| # - name: Install dependencies | |
| # run: | | |
| # if [ -f requirements-dev.txt ]; then pip install -r requirements-dev.txt; fi | |
| # pip install jax==0.3.25 | |
| # pip install jaxlib==0.3.25 | |
| # pip uninstall brainpy -y | |
| # python setup.py install | |
| # - name: Test with pytest | |
| # run: | | |
| # pytest tests/ | |
| test_macos: | |
| runs-on: macos-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| python-version: [ "3.9", "3.10", "3.11"] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Install dependencies | |
| run: | | |
| if [ -f requirements-dev.txt ]; then pip install -r requirements-dev.txt; fi | |
| pip uninstall brainpy -y | |
| python setup.py install | |
| - name: Test with pytest | |
| run: | | |
| pytest tests/ | |
| # test_macos_py37: | |
| # runs-on: macos-latest | |
| # if: github.event.pull_request.merged == true | |
| # strategy: | |
| # fail-fast: false | |
| # matrix: | |
| # python-version: [ "3.7" ] | |
| # | |
| # steps: | |
| # - uses: actions/checkout@v4 | |
| # - name: Set up Python ${{ matrix.python-version }} | |
| # uses: actions/setup-python@v5 | |
| # with: | |
| # python-version: ${{ matrix.python-version }} | |
| # - name: Install dependencies | |
| # run: | | |
| # if [ -f requirements-dev.txt ]; then pip install -r requirements-dev.txt; fi | |
| # pip install jax==0.3.25 | |
| # pip install jaxlib==0.3.25 | |
| # pip uninstall brainpy -y | |
| # python setup.py install | |
| # - name: Test with pytest | |
| # run: | | |
| # pytest tests/ | |
| test_windows: | |
| runs-on: windows-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| python-version: ["3.9", "3.10", "3.11"] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install numpy>=1.21.0 | |
| python -m pip install -r requirements-dev.txt | |
| python -m pip install tqdm brainpylib | |
| pip uninstall brainpy -y | |
| python setup.py install | |
| - name: Test with pytest | |
| run: | | |
| pytest tests/ | |
| # test_windows_py37: | |
| # runs-on: windows-latest | |
| # if: github.event.pull_request.merged == true | |
| # strategy: | |
| # fail-fast: false | |
| # matrix: | |
| # python-version: ["3.7"] | |
| # | |
| # steps: | |
| # - uses: actions/checkout@v4 | |
| # - name: Set up Python ${{ matrix.python-version }} | |
| # uses: actions/setup-python@v5 | |
| # with: | |
| # python-version: ${{ matrix.python-version }} | |
| # - name: Install dependencies | |
| # run: | | |
| # python -m pip install numpy>=1.21.0 | |
| # python -m pip install "jaxlib==0.3.25" -f https://whls.blob.core.windows.net/unstable/index.html --use-deprecated legacy-resolver | |
| # python -m pip install https://github.com/google/jax/archive/refs/tags/jax-v0.3.25.tar.gz | |
| # python -m pip install -r requirements-dev.txt | |
| # python -m pip install tqdm brainpylib | |
| # pip uninstall brainpy -y | |
| # python setup.py install | |
| # - name: Test with pytest | |
| # run: | | |
| # pytest tests/ |