Address comments in PR

Author: mdboom

cuda_core/cuda/core/system/_mig.pxi

@@ -112,7 +112,7 @@ cdef class MigInfo:
         Device
             The parent GPU device for this MIG device.
         """
-        parent_handle = nvml.device_get_handle_from_mig_device_handle(self._device._handle)
+        parent_handle = nvml.device_get_device_handle_from_mig_device_handle(self._device._handle)
         parent_device = Device.__new__(Device)
         parent_device._handle = parent_handle
         return parent_device

cuda_core/tests/system/test_system_device.py

@@ -738,16 +738,15 @@ def test_mig():
             mig = device.mig
 
             assert isinstance(mig.is_mig_device, bool)
-            if mig.is_mig_device:
-                assert isinstance(mig.mode, bool)
-                assert isinstance(mig.pending_mode, bool)
+            assert isinstance(mig.mode, bool)
+            assert isinstance(mig.pending_mode, bool)
 
-                device_count = mig.get_device_count()
-                assert isinstance(device_count, int)
-                assert device_count >= 0
+            device_count = mig.device_count
+            assert isinstance(device_count, int)
+            assert device_count >= 0
 
-                for mig_device in mig.get_all_devices():
-                    assert isinstance(mig_device, system.Device)
+            for mig_device in mig.get_all_devices():
+                assert isinstance(mig_device, system.Device)
 
 
 def test_uuid():