Add flax deps

JaxGaussianProcesses · Apr 7, 2023 · fdff318 · fdff318
1 parent 070dda3
commit fdff318
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -22,6 +22,7 @@ jobs:
           pip install -e .
           pip install -e .[dev]
           pytest --cov=./ --cov-report=xml
+
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v1
         with:

diff --git a/setup.py b/setup.py
@@ -33,6 +33,7 @@ def get_versions():
         "pytest",
         "networkx",
         "pytest-cov",
+        "flax"
     ],
     "cuda": ["jax[cuda]"],
 }