Update test_memory_usage.py

icfaust · web-flow · commit 092e8798dacd · 2025-12-09T22:37:21.000+01:00
diff --git a/sklearnex/tests/test_memory_usage.py b/sklearnex/tests/test_memory_usage.py
@@ -320,7 +320,7 @@ def test_table_conversions_memory_leaks(dataframe, queue, order, data_shape, dty
         and queue.sycl_device.is_gpu
         and (
             os.getenv("ZES_ENABLE_SYSMAN") is None
-            or not is_dpctl_device_available(["gpu"])
+            or not is_sycl_device_available(["gpu"])
         )
     ):
         pytest.skip("SYCL device memory leak check requires the level zero sysman")

Original file line number	Diff line number	Diff line change
`@@ -320,7 +320,7 @@ def test_table_conversions_memory_leaks(dataframe, queue, order, data_shape, dty`
`320`	`320`	`and queue.sycl_device.is_gpu`
`321`	`321`	`and (`
`322`	`322`	`os.getenv("ZES_ENABLE_SYSMAN") is None`
`323`		`- or not is_dpctl_device_available(["gpu"])`
	`323`	`+ or not is_sycl_device_available(["gpu"])`
`324`	`324`	`)`
`325`	`325`	`):`
`326`	`326`	`pytest.skip("SYCL device memory leak check requires the level zero sysman")`