Skip to content

Commit a9eed13

Browse files
committed
Add strict type check for k keyword
1 parent 71d3c23 commit a9eed13

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

dpnp/dpnp_iface_arraycreation.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3680,8 +3680,7 @@ def tri(
36803680
try:
36813681
_k = operator.index(k)
36823682
except TypeError:
3683-
pass
3684-
if _k is None:
3683+
# pylint: disable=raise-missing-from
36853684
raise TypeError(f"`k` must be a integer data type, but got {type(k)}")
36863685

36873686
sycl_dev = dpnp.get_normalized_queue_device(

dpnp/dpnp_iface_indexing.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2621,6 +2621,12 @@ def triu_indices(
26212621
26222622
"""
26232623

2624+
try:
2625+
k = operator.index(k)
2626+
except TypeError:
2627+
# pylint: disable=raise-missing-from
2628+
raise TypeError(f"`k` must be a integer data type, but got {type(k)}")
2629+
26242630
tri_ = ~dpnp.tri(
26252631
n,
26262632
m,

0 commit comments

Comments
 (0)