Skip to content

Commit

Permalink
update jaxconfig import
Browse files Browse the repository at this point in the history
katosh committed Feb 28, 2024
1 parent 7c13864 commit 3ec1e8f
Showing 2 changed files with 2 additions and 1 deletion.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -4,6 +4,7 @@
- Implement `mellon.cov.Linear` covariance kernel
- Change logging setup to configuration dict
- allow setting `active_dims` for composit kernels, allowing more flexible covariance kernel specifications
- update jaxconfig impot for compatibility with newer jax versions

# v1.4.1

2 changes: 1 addition & 1 deletion mellon/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import logging.config
import sys
from jax.config import config as jaxconfig
from jax import config as jaxconfig

from .model import (
DensityEstimator,

0 comments on commit 3ec1e8f

Please sign in to comment.