Skip to content

Commit

Permalink
fixed repetition of average function in docs
Browse files Browse the repository at this point in the history
  • Loading branch information
fnhirwa committed Jan 4, 2023
1 parent 711c3da commit 9b346a8
Showing 1 changed file with 1 addition and 10 deletions.
11 changes: 1 addition & 10 deletions jax/_src/numpy/reductions.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
import builtins
from functools import partial
import operator
from typing import overload, Any, Callable, Literal, Optional, Sequence, Tuple, Union
from typing import Any, Callable, Optional, Sequence, Tuple, Union
import warnings

import numpy as np
Expand Down Expand Up @@ -340,15 +340,6 @@ def _mean(a: ArrayLike, axis: Axis = None, dtype: DTypeLike = None,
sum(a, axis, dtype=dtype, keepdims=keepdims, where=where),
lax.convert_element_type(normalizer, dtype))

@overload
def average(a: ArrayLike, axis: Axis = None, weights: Optional[ArrayLike] = None,
returned: Literal[False] = False, keepdims: bool = False) -> Array: ...
@overload
def average(a: ArrayLike, axis: Axis = None, weights: Optional[ArrayLike] = None, *,
returned: Literal[True], keepdims: bool = False) -> Array: ...
@overload
def average(a: ArrayLike, axis: Axis = None, weights: Optional[ArrayLike] = None,
returned: bool = False, keepdims: bool = False) -> Union[Array, Tuple[Array, Array]]: ...
@_wraps(np.average)
def average(a: ArrayLike, axis: Axis = None, weights: Optional[ArrayLike] = None,
returned: bool = False, keepdims: bool = False) -> Union[Array, Tuple[Array, Array]]:
Expand Down

0 comments on commit 9b346a8

Please sign in to comment.