Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Sep 24, 2024
1 parent 298ed95 commit b3f0dd7
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions source/tests/consistent/test_type_embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,8 +40,6 @@
else:
TypeEmbedNetJAX = object
if INSTALLED_ARRAY_API_STRICT:
import array_api_strict

from ..array_api_strict.utils.type_embed import TypeEmbedNet as TypeEmbedNetStrict
else:
TypeEmbedNetStrict = None
Expand Down Expand Up @@ -133,8 +131,7 @@ def eval_jax(self, jax_obj: Any) -> Any:
def eval_array_api_strict(self, array_api_strict_obj: Any) -> Any:
out = array_api_strict_obj()
return [
np.asarray(x) if hasattr(x, "__array_namespace__") else x
for x in (out,)
np.asarray(x) if hasattr(x, "__array_namespace__") else x for x in (out,)
]

def extract_ret(self, ret: Any, backend) -> Tuple[np.ndarray, ...]:
Expand Down

0 comments on commit b3f0dd7

Please sign in to comment.