Dev

.github/workflows/ci.yml

@@ -41,10 +41,12 @@ jobs:
         run: |
           nox
 
-      - name: Upload coverage reports to Codecov
-        uses: codecov/codecov-action@v3
+      - name: Coveralls
         env:
-          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
+            GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+            pip install coveralls
+            coveralls --service=github
 
   check:
     if: always()

README.md

@@ -1,10 +1,10 @@
 # pynajax
 
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/flatironinstitute/nemos/blob/main/LICENSE)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/pynapple-org/pynajax/blob/main/LICENSE)
 ![Python version](https://img.shields.io/badge/python-3.10-blue.svg)
-[![Project Status: WIP – Initial development is in progress, but there has not yet been a stable, usable release suitable for the public.](https://www.repostatus.org/badges/latest/wip.svg)](https://www.repostatus.org/#wip)
+[![Project Status: Active – The project has reached a stable, usable state and is being actively developed.](https://www.repostatus.org/badges/latest/active.svg)](https://www.repostatus.org/#active)
 [![pynajax CI](https://github.com/pynapple-org/pynajax/actions/workflows/ci.yml/badge.svg)](https://github.com/pynapple-org/pynajax/actions/workflows/ci.yml)
-
+[![Coverage Status](https://coveralls.io/repos/github/pynapple-org/pynajax/badge.svg?branch=main)](https://coveralls.io/github/pynapple-org/pynajax?branch=main)
 
 Welcome to `pynajax`, a GPU accelerated backend for [pynapple](https://github.com/pynapple-org/pynapple) built on top on [jax](https://github.com/google/jax). It offers a fast acceleration for the core pynapple functions using GPU. 
 
@@ -26,6 +26,13 @@ Run the following `pip` command in your virtual environment.
  python -m pip install pynajax
  ```
 
+Alternatively, you can install pynapple and pynajax together.
+
+```bash
+ pip install pynapple[jax]
+```
+
+
 
 ## Basic usage
 

docs/examples/plot_benchmark_event_trigger_average.py

@@ -26,7 +26,7 @@ def get_mean_perf_class(tsg, tsd, ep, binsize):
 
 def benchmark_event_trigger_average(binsize):
     times = []
-    for T in np.arange(500000, 800000, 100000):
+    for T in np.arange(100000, 1000000, 100000):
         time_array = np.arange(T) / 2
         data_array = np.arange(16 * T).reshape(T, 4, 2, 2)
         starts = np.arange(1, T // 2 - 1, 20)

docs/index.md

@@ -2,8 +2,9 @@
 
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/pynapple-org/pynajax/blob/main/LICENSE)
 ![Python version](https://img.shields.io/badge/python-3.10-blue.svg)
-[![Project Status: WIP – Initial development is in progress, but there has not yet been a stable, usable release suitable for the public.](https://www.repostatus.org/badges/latest/wip.svg)](https://www.repostatus.org/#wip)
+[![Project Status: Active – The project has reached a stable, usable state and is being actively developed.](https://www.repostatus.org/badges/latest/active.svg)](https://www.repostatus.org/#active)
 [![pynajax CI](https://github.com/pynapple-org/pynajax/actions/workflows/ci.yml/badge.svg)](https://github.com/pynapple-org/pynajax/actions/workflows/ci.yml)
+[![Coverage Status](https://coveralls.io/repos/github/pynapple-org/pynajax/badge.svg?branch=main)](https://coveralls.io/github/pynapple-org/pynajax?branch=main)
 
 Welcome to `pynajax`, a GPU accelerated backend for [pynapple](https://github.com/pynapple-org/pynapple) built on top on [jax](https://github.com/google/jax). It offers a fast acceleration for the core pynapple functions using GPU. 
 
@@ -26,6 +27,12 @@ Run the following `pip` command in your virtual environment.
  python -m pip install pynajax
  ```
 
+Alternatively, you can install pynapple and pynajax together.
+
+```bash
+ pip install pynapple[jax]
+```
+
 ## Basic usage
 
 To use pynajax, you need to change the pynapple backend using `nap.nap_config.set_backend`. See the example below :