1.4.8 - fixes jax bug.

CODES-group · Mar 21, 2024 · 6e56297 · 6e56297
1 parent 43f6bc8
commit 6e56297
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/environment.yml b/environment.yml
@@ -9,7 +9,7 @@ dependencies:
   - numpy>=1.24.1
   - scipy>=1.10.0
   - tqdm>=4.64.0
-  - matplotlib>=3.6.2
+  - matplotlib>=3.6.2, <=3.8
   - jax>=0.4.0
   - cvxopt>=1.2.7
   - cyipopt>=1.2.0

diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "opyrability"
-version = "1.4.7"
+version = "1.4.8"
 description = "Process operability analysis in Python"
 authors = ["Victor Alves"]
 license = "MIT License"

diff --git a/setup.py b/setup.py
@@ -17,7 +17,7 @@
 
 setup_kwargs = {
     'name': 'opyrability',
-    'version': '1.4.7',
+    'version': '1.4.8',
     'description': 'Process operability analysis in Python',
     'author': 'Victor Alves',
     'author_email': 'None',

diff --git a/src/opyrability.py b/src/opyrability.py
@@ -1816,6 +1816,8 @@ def F_io(o, i): return model(o, i)
     if derivative == 'jax':
         from jax import config
         config.update("jax_enable_x64", True)
+        config.update('jax_platform_name', 'cpu')
+        warnings.filterwarnings('ignore', module='jax._src.lib.xla_bridge')
         import jax.numpy as jnp
         from jax import jit, jacrev
         from jax.experimental.ode import odeint as odeint