fix wrong type set

ModelCloud · Jan 8, 2025 · a505143 · a505143
1 parent 0627602
commit a505143
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/device_smi/amd.py b/device_smi/amd.py
@@ -10,7 +10,7 @@ class AMDGPUMetrics(BaseMetrics):
 
 class AMDDevice(GPUDevice):
     def __init__(self, cls, index):
-        super().__init__(cls, index, "xpu")
+        super().__init__(cls, index)
         self.gpu_id = self._get_gpu_id()
 
         try:

diff --git a/device_smi/intel.py b/device_smi/intel.py
@@ -10,7 +10,7 @@ class IntelGPUMetrics(BaseMetrics):
 
 class IntelDevice(GPUDevice):
     def __init__(self, cls, index: int = 0):
-        super().__init__(cls, index)
+        super().__init__(cls, index, "xpu")
         self.gpu_id = index
 
         try: