Dependency updates

.github/workflows/test-bt_ocean.yml

@@ -27,7 +27,7 @@ jobs:
         run: |
           python3 -m venv bt_ocean_venv
           . bt_ocean_venv/bin/activate
-          python3 -m pip install flake8 jax[cpu] keras matplotlib pytest pytest-timeout pytest-xdist ruff scipy sympy zarr
+          python3 -m pip install flake8 jax[cpu] keras matplotlib numpy pytest pytest-timeout pytest-xdist ruff scipy sympy zarr
       - name: Lint
         run: |
           . bt_ocean_venv/bin/activate

bt_ocean/__init__.py

@@ -1,6 +1,12 @@
 def _init():
+    import keras
+
     import logging
     import sys
+    import warnings
+
+    if keras.backend.backend() != "jax":
+        warnings.warn("bt_ocean requires Keras with the JAX backend", ImportWarning)
 
     logger = logging.getLogger("bt_ocean")
 

pyproject.toml

@@ -15,9 +15,17 @@ dependencies = [
     "keras",
     "numpy",
     "scipy",
-    "sympy",
     "zarr"
 ]
+optional-dependencies.test = [
+    "matplotlib",
+    "pytest",
+    "sympy"
+]
+optional-dependencies.lint = [
+    "flake8",
+    "ruff"
+]
 
 [tool.ruff]