Skip to content

Commit

Permalink
Black linting
Browse files Browse the repository at this point in the history
  • Loading branch information
aarmey committed Aug 1, 2024
1 parent 3b861da commit d452148
Show file tree
Hide file tree
Showing 2 changed files with 61 additions and 57 deletions.
116 changes: 60 additions & 56 deletions tensorly/backend/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,62 +32,66 @@ def __get__(self, instance, cls=None):


class BackendManager(types.ModuleType):
_functions = [
"moveaxis",
"trace",
"shape",
"ndim",
"where",
"copy",
"transpose",
"arange",
"eye",
"kron",
"concatenate",
"max",
"all",
"mean",
"sum",
"prod",
"sign",
"argmin",
"argmax",
"stack",
"conj",
"diag",
"dot",
"tensordot",
"clip",
"kron",
"diagonal",
"lstsq",
"eps",
"finfo",
"solve",
"svd",
"qr",
"randn",
"gamma",
"digamma",
"check_random_state",
"sort",
"eigh",
"index_update",
"context",
"tensor",
"norm",
"to_numpy",
"is_tensor",
"argsort",
"flip",
"asin",
"acos",
"atan",
"asinh",
"acosh",
"atanh",
"logsumexp",
] + backend_array + backend_basic_math
_functions = (
[
"moveaxis",
"trace",
"shape",
"ndim",
"where",
"copy",
"transpose",
"arange",
"eye",
"kron",
"concatenate",
"max",
"all",
"mean",
"sum",
"prod",
"sign",
"argmin",
"argmax",
"stack",
"conj",
"diag",
"dot",
"tensordot",
"clip",
"kron",
"diagonal",
"lstsq",
"eps",
"finfo",
"solve",
"svd",
"qr",
"randn",
"gamma",
"digamma",
"check_random_state",
"sort",
"eigh",
"index_update",
"context",
"tensor",
"norm",
"to_numpy",
"is_tensor",
"argsort",
"flip",
"asin",
"acos",
"atan",
"asinh",
"acosh",
"atanh",
"logsumexp",
]
+ backend_array
+ backend_basic_math
)
_attributes = [
"nan",
"index",
Expand Down
2 changes: 1 addition & 1 deletion tensorly/backend/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -1252,7 +1252,7 @@ def acosh(self, x):
def atanh(self, x):
"""Return the arctanh of x."""
return self.arctanh(x)

@staticmethod
def diagonal(tensor, offset=0, axis1=0, axis2=1):
"""
Expand Down

0 comments on commit d452148

Please sign in to comment.