Skip to content

Commit 69557ee

Browse files
authored
Fix "ufunc 'isfinite' not supported for the input types..." part 1 (#932)
1 parent 38d42dc commit 69557ee

File tree

3 files changed

+10
-7
lines changed

3 files changed

+10
-7
lines changed

dpnp/dpnp_iface_logic.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@
4343
import numpy
4444
import dpnp
4545

46+
import dpnp.config as config
4647
from dpnp.dpnp_algo import *
4748
from dpnp.dpnp_utils import *
4849

@@ -380,6 +381,8 @@ def isclose(x1, x2, rtol=1e-05, atol=1e-08, equal_nan=False):
380381
[True, False]
381382
382383
"""
384+
if config.__DPNP_OUTPUT_DPCTL__:
385+
return call_origin(numpy.isclose, x1, x2, rtol=rtol, atol=atol, equal_nan=equal_nan)
383386

384387
x1_desc = dpnp.get_dpnp_descriptor(x1)
385388
x2_desc = dpnp.get_dpnp_descriptor(x2)

dpnp/dpnp_iface_mathematical.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -428,7 +428,7 @@ def cross(x1, x2, axisa=-1, axisb=-1, axisc=-1, axis=None):
428428
elif axis is not None:
429429
pass
430430
else:
431-
return dpnp_cross(x1_desc, x2_desc)
431+
return dpnp_cross(x1_desc, x2_desc).get_pyobj()
432432

433433
return call_origin(numpy.cross, x1, x2, axisa, axisb, axisc, axis)
434434

tests/test_random.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ def check_seed(self, dist_name, params):
3535
seed = 28041990
3636
size = 10
3737
dpnp.random.seed(seed)
38-
a1 = numpy.asarray(getattr(dpnp.random, dist_name)(size=size, **params))
38+
a1 = dpnp.asarray(getattr(dpnp.random, dist_name)(size=size, **params))
3939
dpnp.random.seed(seed)
40-
a2 = numpy.asarray(getattr(dpnp.random, dist_name)(size=size, **params))
40+
a2 = dpnp.asarray(getattr(dpnp.random, dist_name)(size=size, **params))
4141
assert_allclose(a1, a2, rtol=1e-07, atol=0)
4242

4343

@@ -638,9 +638,9 @@ def test_seed(self, df):
638638
size = 10
639639
nonc = 1.8
640640
dpnp.random.seed(seed)
641-
a1 = numpy.asarray(dpnp.random.noncentral_chisquare(df, nonc, size=size))
641+
a1 = dpnp.asarray(dpnp.random.noncentral_chisquare(df, nonc, size=size))
642642
dpnp.random.seed(seed)
643-
a2 = numpy.asarray(dpnp.random.noncentral_chisquare(df, nonc, size=size))
643+
a2 = dpnp.asarray(dpnp.random.noncentral_chisquare(df, nonc, size=size))
644644
assert_allclose(a1, a2, rtol=1e-07, atol=0)
645645

646646

@@ -877,9 +877,9 @@ def test_seed(self, kappa):
877877
size = 10
878878
mu = 2.
879879
dpnp.random.seed(seed)
880-
a1 = numpy.asarray(dpnp.random.vonmises(mu, kappa, size=size))
880+
a1 = dpnp.asarray(dpnp.random.vonmises(mu, kappa, size=size))
881881
dpnp.random.seed(seed)
882-
a2 = numpy.asarray(dpnp.random.vonmises(mu, kappa, size=size))
882+
a2 = dpnp.asarray(dpnp.random.vonmises(mu, kappa, size=size))
883883
assert_allclose(a1, a2, rtol=1e-07, atol=0)
884884

885885

0 commit comments

Comments
 (0)