Tweak create_sub_devices docstring and raise TypeError if partition i…

…s not specified
IntelPython · Jan 13, 2025 · 9559620 · 9559620
1 parent ec6c1d9
commit 9559620
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/dpctl/_sycl_device.pyx b/dpctl/_sycl_device.pyx
@@ -1660,14 +1660,13 @@ cdef class SyclDevice(_SyclDevice):
 
         Raises:
             ValueError:
-                If the ``partition`` keyword argument is not specified or
-                the affinity domain string is not legal or is not one of the
-                three supported options.
+                If the ``partition`` keyword argument is not one of the three
+                supported options or is not a legal affinity domain string.
             dpctl.SyclSubdeviceCreationError:
                 If sub-devices can not be created.
         """
         if "partition" not in kwargs:
-            raise ValueError(
+            raise TypeError(
                 "create_sub_devices(partition=parition_spec) is expected."
             )
         partition = kwargs.pop("partition")