Add pytest instructions and GitHub action (#6)

Author: jeffcarp

.github/workflows/test.yml

@@ -0,0 +1,20 @@
+name: Test
+on: [push, pull_request]
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.x'
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -r requirements.txt
+    - name: Test with pytest
+      run: |
+        pip install pytest
+        pytest ./src/
+

README.md

@@ -1,6 +1,6 @@
 # metrax
 
-`metrax` is an official JAX metrics library.
+`metrax` is a library with standard eval metrics implementations in JAX.
 
 ## Installation
 
@@ -12,6 +12,12 @@ pip install google-metrax
 
 ## Development
 
+Run the tests:
+
+```sh
+pytest src/metrax
+```
+
 Develop the docs locally:
 
 ```

requirements.txt

@@ -0,0 +1,2 @@
+clu==0.0.12
+scikit-learn==1.6.1