gpjax 0.13.4__py3-none-any.whl → 0.13.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. gpjax/__init__.py +1 -1
  2. gpjax/parameters.py +2 -1
  3. {gpjax-0.13.4.dist-info → gpjax-0.13.5.dist-info}/METADATA +3 -2
  4. {gpjax-0.13.4.dist-info → gpjax-0.13.5.dist-info}/RECORD +6 -6
  5. {gpjax-0.13.4.dist-info → gpjax-0.13.5.dist-info}/WHEEL +1 -1
  6. {gpjax-0.13.4.dist-info → gpjax-0.13.5.dist-info}/licenses/LICENSE.txt +0 -0
gpjax/__init__.py CHANGED
@@ -40,7 +40,7 @@ __license__ = "MIT"
40
40
  __description__ = "Gaussian processes in JAX and Flax"
41
41
  __url__ = "https://github.com/thomaspinder/GPJax"
42
42
  __contributors__ = "https://github.com/thomaspinder/GPJax/graphs/contributors"
43
- __version__ = "0.13.4"
43
+ __version__ = "0.13.5"
44
44
 
45
45
  __all__ = [
46
46
  "gps",
gpjax/parameters.py CHANGED
@@ -1,6 +1,7 @@
1
1
  import typing as tp
2
2
 
3
3
  from flax import nnx
4
+ import jax
4
5
  from jax.experimental import checkify
5
6
  import jax.numpy as jnp
6
7
  import jax.tree_util as jtu
@@ -162,7 +163,7 @@ def _check_is_arraylike(value: T) -> None:
162
163
  Raises:
163
164
  TypeError: If the value is not array-like.
164
165
  """
165
- if not isinstance(value, (ArrayLike, list)):
166
+ if not isinstance(value, (jax.Array, ArrayLike, list)):
166
167
  raise TypeError(
167
168
  f"Expected parameter value to be an array-like type. Got {value}."
168
169
  )
{gpjax-0.13.4.dist-info → gpjax-0.13.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: gpjax
3
- Version: 0.13.4
3
+ Version: 0.13.5
4
4
  Summary: Gaussian processes in JAX.
5
5
  Project-URL: Documentation, https://docs.jaxgaussianprocesses.com/
6
6
  Project-URL: Issues, https://github.com/thomaspinder/GPJax/issues
@@ -25,6 +25,7 @@ Requires-Dist: jaxtyping>0.2.10
25
25
  Requires-Dist: numpy>=2.0.0
26
26
  Requires-Dist: numpyro
27
27
  Requires-Dist: optax>0.2.1
28
+ Requires-Dist: tensorstore!=0.1.76; sys_platform == 'darwin'
28
29
  Requires-Dist: tqdm>4.66.2
29
30
  Provides-Extra: dev
30
31
  Requires-Dist: absolufy-imports>=0.3.1; extra == 'dev'
@@ -59,7 +60,7 @@ Requires-Dist: mkdocs-jupyter>=0.24.3; extra == 'docs'
59
60
  Requires-Dist: mkdocs-literate-nav>=0.6.0; extra == 'docs'
60
61
  Requires-Dist: mkdocs-material>=9.5.12; extra == 'docs'
61
62
  Requires-Dist: mkdocs>=1.5.3; extra == 'docs'
62
- Requires-Dist: mkdocstrings[python]<0.31.0; extra == 'docs'
63
+ Requires-Dist: mkdocstrings[python]<1.1.0; extra == 'docs'
63
64
  Requires-Dist: nbconvert>=7.16.2; extra == 'docs'
64
65
  Requires-Dist: networkx>=3.0; extra == 'docs'
65
66
  Requires-Dist: pandas>=1.5.3; extra == 'docs'
{gpjax-0.13.4.dist-info → gpjax-0.13.5.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- gpjax/__init__.py,sha256=88sVAK1NfRp73-rINsqz8sWUQe4pVF_avvJHjfCu-Is,1625
1
+ gpjax/__init__.py,sha256=8guaWVHiNXlZ9m_XvjPo8nherGpCpGWCCxB5fenq_Gw,1625
2
2
  gpjax/citation.py,sha256=9L-OVIDCrwjM3TeWinC5UHS3Lxn_9sVIsj_H9i08BT8,4428
3
3
  gpjax/dataset.py,sha256=Ef5JGrl4jJS1mQmL3JdO0fdqbVmflT_Cu5VrlpYdJY4,4071
4
4
  gpjax/distributions.py,sha256=iKmeQ_NN2CIjRiuOeJlwEGASzGROi4ZCerVi1uY7zRM,7758
@@ -9,7 +9,7 @@ gpjax/likelihoods.py,sha256=SzGh3uAkULVMfAVA3QpKQwWp-XAcx46B7QhhgKymn6A,17381
9
9
  gpjax/mean_functions.py,sha256=cDv3X5E4-to8pgzHpbKhoMKDLdWg3WhEq1W21kvbDDg,6571
10
10
  gpjax/numpyro_extras.py,sha256=-vWJ7SpZVNhSdCjjrlxIkovMFrM1IzpsMJK3B4LioGE,3411
11
11
  gpjax/objectives.py,sha256=QScq5e77-jugtt2wKZNLgk3VtkqSDddJyYBs-hQ7dZc,17206
12
- gpjax/parameters.py,sha256=BS1I6MluWi2CGggk-ikA5STJmHJiQSZ8zro3laAkwTI,7013
12
+ gpjax/parameters.py,sha256=iGab7IMAANuhFm8ojcK5x5fJNzUOdflr2h2pBaVLe9Q,7035
13
13
  gpjax/scan.py,sha256=Z_V1yd76dL4B7_rnJnr_fohom6xzN_WRYxlTAAdqfa0,5357
14
14
  gpjax/typing.py,sha256=M3CvWsYtZ3PFUvBvvbRNjpwerNII0w4yGuP0I-sLeYI,1705
15
15
  gpjax/variational_families.py,sha256=J7L1lh0McYSgr1BKq5vsVjtfu2XeyOCn-I_AHQym-xA,36723
@@ -46,7 +46,7 @@ gpjax/linalg/__init__.py,sha256=F8mxk_9Zc2nFd7Q-unjJ50_6rXEKzZj572WsU_jUKqI,547
46
46
  gpjax/linalg/operations.py,sha256=xvhOy5P4FmUCPWjIVNdg1yDXaoFQ48anFUfR-Tnfr6k,6480
47
47
  gpjax/linalg/operators.py,sha256=arxRGwcoAy_RqUYqBpZ3XG6OXbjShUl7m8sTpg85npE,11608
48
48
  gpjax/linalg/utils.py,sha256=fKV8G_iKZVhNkNvN20D_dQEi93-8xosGbXBP-v7UEyo,2020
49
- gpjax-0.13.4.dist-info/METADATA,sha256=UOQ_9_Yj0_qnh-nDZljelyArni74kTaHvvowfc5t918,10322
50
- gpjax-0.13.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
51
- gpjax-0.13.4.dist-info/licenses/LICENSE.txt,sha256=3umwi0h8wmKXOZO8XwRBwSl3vJt2hpWKEqSrSXLR7-I,1084
52
- gpjax-0.13.4.dist-info/RECORD,,
49
+ gpjax-0.13.5.dist-info/METADATA,sha256=qJ8ZeG1TGMX6a_2WQnNSwTl2JBXVp8107QqY9f1XWyE,10382
50
+ gpjax-0.13.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
51
+ gpjax-0.13.5.dist-info/licenses/LICENSE.txt,sha256=3umwi0h8wmKXOZO8XwRBwSl3vJt2hpWKEqSrSXLR7-I,1084
52
+ gpjax-0.13.5.dist-info/RECORD,,
{gpjax-0.13.4.dist-info → gpjax-0.13.5.dist-info}/WHEEL RENAMED
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: hatchling 1.27.0
2
+ Generator: hatchling 1.28.0
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any