Skip to content

Commit bff9bf2

Browse files
committed
Fix ruff warnings
1 parent 244462f commit bff9bf2

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

array_api_compat/common/_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
if TYPE_CHECKING:
1313
from typing import Optional, Union, Any
14-
from ._typing import Array, Device
14+
from ._typing import Array, Device
1515

1616
import sys
1717
import math
@@ -277,7 +277,7 @@ def to_device(x: Array, device: Device, /, *, stream: Optional[Union[int, Any]]
277277
raise ValueError(f"Unsupported device {device!r}")
278278
elif is_jax_array(x):
279279
# This import adds to_device to x
280-
import jax.experimental.array_api
280+
import jax.experimental.array_api # noqa: F401
281281
return x.to_device(device, stream=stream)
282282
return x.to_device(device, stream=stream)
283283

tests/test_common.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
from array_api_compat import (is_numpy_array, is_cupy_array, is_torch_array,
2-
is_dask_array, is_jax_array, is_array_api_obj,
3-
device, to_device)
1+
from array_api_compat import (is_numpy_array, is_cupy_array, is_torch_array, # noqa: F401
2+
is_dask_array, is_jax_array)
3+
4+
from array_api_compat import is_array_api_obj, device, to_device
45

56
from ._helpers import import_
67

0 commit comments

Comments
 (0)