The annotation of the device that has been deleted is incorrect. (#323)

Signed-off-by: chaunceyjiang <chaunceyjiang@gmail.com>
Project-HAMi · May 28, 2024 · 4af7069 · 4af7069
1 parent 18e86e5
commit 4af7069
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/device/hygon/device.go b/pkg/device/hygon/device.go
@@ -128,7 +128,7 @@ func (dev *DCUDevices) GetNodeDevices(n corev1.Node) ([]*api.DeviceInfo, error)
 }
 
 func (dev *DCUDevices) NodeCleanUp(nn string) error {
-	return util.MarkAnnotationsToDelete(HygonDCUDevice, nn)
+	return util.MarkAnnotationsToDelete(HandshakeAnnos, nn)
 }
 
 func (dev *DCUDevices) CheckHealth(devType string, n *corev1.Node) (bool, bool) {

diff --git a/pkg/device/nvidia/device.go b/pkg/device/nvidia/device.go
@@ -78,7 +78,7 @@ func (dev *NvidiaGPUDevices) ParseConfig(fs *flag.FlagSet) {
 }
 
 func (dev *NvidiaGPUDevices) NodeCleanUp(nn string) error {
-	return util.MarkAnnotationsToDelete(NvidiaGPUDevice, nn)
+	return util.MarkAnnotationsToDelete(HandshakeAnnos, nn)
 }
 
 func (dev *NvidiaGPUDevices) CheckHealth(devType string, n *corev1.Node) (bool, bool) {