mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support older Numpy versions that don't have np.quantile.

Browse Source 
Fix typo.
This commit is contained in:
Peter Hawkins 2019-07-20 08:44:04 +01:00
parent e71680212f
commit eefd551767
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/numpy/lax_numpy.py
View File

@ -2486,7 +2486,7 @@ def cov(m, y=None, rowvar=True, bias=False, ddof=None, fweights=None,
return true_divide(dot(X, X_T.conj()), f).squeeze()
@_wraps(onp.quantile)
@_wraps(getattr(onp, "quantile", None))
def quantile(a, q, axis=None, out=None, overwrite_input=False,
interpolation="linear", keepdims=False):
if overwrite_input or out is not None:

2
jax/tree_util.py
View File

@ -259,7 +259,7 @@ NamedtupleNode = NodeType('namedtuple',
class Partial(functools.partial):
"""A version of functools.partial that works in pytrees.
Use it for partial function evaluation in a way that is compatibile with JAX's
Use it for partial function evaluation in a way that is compatible with JAX's
transformations, e.g., ``Partial(func, *args, **kwargs)``.
(You need to explicitly opt-in to this behavior because we didn't want to give