assert base attrs are valid

device_smi/device.py

@@ -19,7 +19,7 @@
 
 class Device:
     def __init__(self, device):
-
+        # init attribute first to avoid IDE not attr warning
         # CPU/GPU Device
         self.memory_total = None
         self.type = None
@@ -35,6 +35,9 @@ def __init__(self, device):
             device_index = device.index
         elif f"{device}".lower() == "os":
             self.device = OSDevice(self)
+            assert self.arch
+            assert self.version
+            assert self.name
             return
         else:
             d = f"{device}".lower()
@@ -79,6 +82,12 @@ def __init__(self, device):
         else:
             raise Exception(f"The device {device_type} is not supported")
 
+        assert self.memory_total
+        assert self.type
+        assert self.features
+        assert self.vendor
+        assert self.model
+
     def info(self):
         warnings.warn(
             "info() method is deprecated and will be removed in next release.",