missing property

apache · Nov 10, 2023 · da1987a · da1987a
1 parent 7d2db62
commit da1987a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/python/src/nanoarrow/_lib.pyx b/python/src/nanoarrow/_lib.pyx
@@ -967,6 +967,10 @@ cdef class Device:
     def device_type(self):
         return self._ptr.device_type
 
+    @property
+    def device_id(self):
+        return self._ptr.device_id
+
     @staticmethod
     def resolve(ArrowDeviceType device_type, int64_t device_id):
         if device_type == ARROW_DEVICE_CPU:

diff --git a/python/tests/test_nanoarrow_device.py b/python/tests/test_nanoarrow_device.py
@@ -24,6 +24,7 @@
 def test_cpu_device():
     cpu = na._lib.Device.cpu()
     assert cpu.device_type == 1
+    assert cpu.device_id == 0
 
     cpu = na._lib.Device.resolve(1, 0)
     assert cpu.device_type == 1