Skip to content

Commit 3ee2076

Browse files
Apply remarks
1 parent 1ee5b5a commit 3ee2076

File tree

2 files changed

+35
-26
lines changed

2 files changed

+35
-26
lines changed

dpnp/__init__.py

Lines changed: 33 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,6 @@
140140
atleast_2d,
141141
atleast_3d,
142142
broadcast_arrays,
143-
broadcast_shapes,
144143
broadcast_to,
145144
can_cast,
146145
column_stack,
@@ -181,18 +180,13 @@
181180
tile,
182181
transpose,
183182
trim_zeros,
184-
unique,
185-
unique_all,
186-
unique_counts,
187-
unique_inverse,
188-
unique_values,
189183
unstack,
190184
vsplit,
191185
vstack,
192186
)
193187

194188
# -----------------------------------------------------------------------------
195-
# Binary operations
189+
# Bit-wise operations
196190
# -----------------------------------------------------------------------------
197191
from .dpnp_iface_bitwise import (
198192
binary_repr,
@@ -210,7 +204,7 @@
210204
)
211205

212206
# -----------------------------------------------------------------------------
213-
# Functional routines
207+
# Functional programming
214208
# -----------------------------------------------------------------------------
215209
from .dpnp_iface_functional import (
216210
apply_along_axis,
@@ -375,9 +369,7 @@
375369
nanprod,
376370
nansum,
377371
)
378-
379372
from .dpnp_iface_statistics import convolve
380-
381373
from .dpnp_iface_trigonometric import (
382374
arccos,
383375
arccosh,
@@ -424,36 +416,42 @@
424416
unwrap,
425417
)
426418

419+
# -----------------------------------------------------------------------------
420+
# Miscellaneous routines
421+
# -----------------------------------------------------------------------------
422+
from .dpnp_iface_manipulation import broadcast_shapes
423+
from .dpnp_iface import get_include
424+
425+
# -----------------------------------------------------------------------------
426+
# Set routines
427+
# -----------------------------------------------------------------------------
428+
from .dpnp_iface_manipulation import (
429+
unique,
430+
unique_all,
431+
unique_counts,
432+
unique_inverse,
433+
unique_values,
434+
)
435+
427436
# -----------------------------------------------------------------------------
428437
# Sorting, searching, and counting
429438
# -----------------------------------------------------------------------------
430439
from .dpnp_iface_counting import count_nonzero
431-
432440
from .dpnp_iface_nanfunctions import nanargmax, nanargmin
433-
434441
from .dpnp_iface_searching import (
435442
argmax,
436443
argmin,
437444
argwhere,
438445
searchsorted,
439446
where,
440447
)
441-
442448
from .dpnp_iface_sorting import (
443449
argsort,
444450
partition,
445451
sort,
446452
sort_complex,
447453
)
448454

449-
from .dpnp_iface_window import (
450-
bartlett,
451-
blackman,
452-
hamming,
453-
hanning,
454-
kaiser,
455-
)
456-
457455
# -----------------------------------------------------------------------------
458456
# Statistics
459457
# -----------------------------------------------------------------------------
@@ -465,7 +463,6 @@
465463
histogram2d,
466464
histogramdd,
467465
)
468-
469466
from .dpnp_iface_nanfunctions import (
470467
nanmax,
471468
nanmean,
@@ -474,7 +471,6 @@
474471
nanstd,
475472
nanvar,
476473
)
477-
478474
from .dpnp_iface_statistics import (
479475
amax,
480476
amin,
@@ -492,7 +488,20 @@
492488
)
493489

494490
# -----------------------------------------------------------------------------
495-
# DPNP iface functions
491+
# Window functions
492+
# -----------------------------------------------------------------------------
493+
from .dpnp_iface_window import (
494+
bartlett,
495+
blackman,
496+
hamming,
497+
hanning,
498+
kaiser,
499+
)
500+
501+
# =============================================================================
502+
503+
# -----------------------------------------------------------------------------
504+
# Helper functions
496505
# -----------------------------------------------------------------------------
497506
from .dpnp_iface import (
498507
are_same_logical_tensors,
@@ -502,7 +511,6 @@
502511
check_supported_arrays_type,
503512
default_float_type,
504513
get_dpnp_descriptor,
505-
get_include,
506514
get_normalized_queue_device,
507515
get_result_array,
508516
get_usm_ndarray,

dpnp/dpnp_iface.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,8 @@
5050
from dpctl.tensor._device import normalize_queue_device
5151

5252
import dpnp
53-
from dpnp.dpnp_array import dpnp_array
53+
54+
from .dpnp_array import dpnp_array
5455

5556
# pylint: disable=no-name-in-module
5657
from .dpnp_utils import (

0 commit comments

Comments
 (0)