Skip to content

Commit

Permalink
Black
Browse files Browse the repository at this point in the history
  • Loading branch information
vshampor committed Oct 27, 2023
1 parent 7a192c2 commit 353fd1c
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions nncf/experimental/tensor/functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,9 +73,7 @@ def flatten(a: Tensor) -> Tensor:

@functools.singledispatch
@_tensor_guard
def max(
a: Tensor, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: Optional[bool] = False
) -> Tensor:
def max(a: Tensor, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: Optional[bool] = False) -> Tensor:
"""
Return the maximum of an array or maximum along an axis.
Expand All @@ -90,9 +88,7 @@ def max(

@functools.singledispatch
@_tensor_guard
def min(
a: Tensor, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: Optional[bool] = False
) -> Tensor:
def min(a: Tensor, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: Optional[bool] = False) -> Tensor:
"""
Return the minimum of an array or minimum along an axis.
Expand Down

0 comments on commit 353fd1c

Please sign in to comment.