Skip to content

Commit cb0ad39

Browse files
MAINT: Stronger check for dpctl availability (#3102)
* stronger check for dpctl availability * remove function that's only used once
1 parent a1b8bc1 commit cb0ad39

File tree

1 file changed

+5
-31
lines changed

1 file changed

+5
-31
lines changed

onedal/utils/_third_party.py

Lines changed: 5 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -28,44 +28,18 @@
2828

2929
from daal4py.sklearn._utils import _package_check_version
3030

31-
32-
@functools.lru_cache(maxsize=256, typed=False)
33-
def is_dpctl_available(version=None):
34-
"""Check availability of DPCtl package.
35-
36-
Parameters
37-
----------
38-
version : str or None, default=None
39-
Minimum supported dpctl version if installed.
40-
Secondary version check skipped if set to None.
41-
42-
Returns
43-
-------
44-
dpctl_available : bool
45-
Flag describing import success.
46-
"""
47-
try:
48-
import dpctl
49-
50-
dpctl_available = True
51-
except ImportError:
52-
dpctl_available = False
53-
if dpctl_available and version is not None:
54-
dpctl_available = _package_check_version(version, dpctl.__version__)
55-
return dpctl_available
56-
57-
5831
# Note: The dpctl package provides SYCL infrastructure (e.g. SyclQueue)
5932
# which is loaded as normal as it is preferred over included backend
6033
# replacements in the core onedal python module.
61-
dpctl_available = is_dpctl_available()
62-
63-
if dpctl_available:
34+
try:
6435
from dpctl import SyclQueue
65-
else:
36+
37+
dpctl_available = True
38+
except ImportError:
6639
from onedal import _default_backend as backend
6740

6841
SyclQueue = backend.SyclQueue
42+
dpctl_available = False
6943

7044

7145
def lazy_import(*module_names: str) -> Callable:

0 commit comments

Comments
 (0)