Skip to content

Commit 5e115bf

Browse files
authored
FIX: minor follow up to #1936 and #2045 (#2144)
1 parent b0cb177 commit 5e115bf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sklearnex/tests/test_memory_usage.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
)
3737
from onedal.tests.utils._device_selection import get_queues, is_dpctl_device_available
3838
from onedal.utils._array_api import _get_sycl_namespace
39-
from onedal.utils._dpep_helpers import dpctl_available, dpnp_available, is_dpctl_available
39+
from onedal.utils._dpep_helpers import dpctl_available, dpnp_available
4040
from sklearnex import config_context
4141
from sklearnex.tests.utils import PATCHED_FUNCTIONS, PATCHED_MODELS, SPECIAL_INSTANCES
4242
from sklearnex.utils._array_api import get_namespace
@@ -365,7 +365,7 @@ def test_table_conversions_memory_leaks(dataframe, queue, order, data_shape, dty
365365
func = ORDER_DICT[order]
366366

367367
if queue.sycl_device.is_gpu and (
368-
os.getenv("ZES_ENABLE_SYSMAN") is None or not is_dpctl_available("gpu")
368+
os.getenv("ZES_ENABLE_SYSMAN") is None or not is_dpctl_device_available("gpu")
369369
):
370370
pytest.skip("SYCL device memory leak check requires the level zero sysman")
371371

0 commit comments

Comments
 (0)