From 0cc1860f8f8359786aa96c53e18d717dd914777d Mon Sep 17 00:00:00 2001
From: CSY <csy@modelcloud.ai>
Date: Wed, 4 Dec 2024 15:39:54 +0800
Subject: [PATCH] assert base attrs are valid

---
 device_smi/device.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/device_smi/device.py b/device_smi/device.py
index 62b11cd..9bef93e 100644
--- a/device_smi/device.py
+++ b/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.",