fix

Signed-off-by: Gyuho Lee <[email protected]>
leptonai · Nov 30, 2024 · 74fd8b4 · 74fd8b4
1 parent 9fad69d
commit 74fd8b4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/client/v1/examples/get-states/main.go b/client/v1/examples/get-states/main.go
@@ -3,7 +3,6 @@ package main
 import (
 	"context"
 	"errors"
-	"fmt"
 	"time"
 
 	client_v1 "github.com/leptonai/gpud/client/v1"
@@ -29,7 +28,7 @@ func main() {
 
 		for _, ss := range states {
 			for _, s := range ss.States {
-				fmt.Printf("state: %q, healthy: %v, extra info: %q\n", s.Name, s.Healthy, s.ExtraInfo)
+				log.Logger.Infof("state: %q, healthy: %v, extra info: %q\n", s.Name, s.Healthy, s.ExtraInfo)
 			}
 		}
 	}

diff --git a/cmd/gpud/command/status.go b/cmd/gpud/command/status.go
@@ -97,10 +97,7 @@ func checkDiskComponent() error {
 	defer cancel()
 	states, err := client.GetStates(ctx, baseURL, client.WithComponent(componentName))
 	if err != nil {
-		if errors.Is(err, errdefs.ErrNotFound) {
-			log.Logger.Warnw("component not found", "component", componentName)
-			return nil
-		}
+		// assume disk component is enabled for all platforms
 		return err
 	}
 	if len(states) == 0 {
@@ -110,7 +107,7 @@ func checkDiskComponent() error {
 
 	for _, ss := range states {
 		for _, s := range ss.States {
-			fmt.Printf("state: %q, healthy: %v, extra info: %q\n", s.Name, s.Healthy, s.ExtraInfo)
+			log.Logger.Infof("state: %q, healthy: %v, extra info: %q\n", s.Name, s.Healthy, s.ExtraInfo)
 		}
 	}
 
@@ -138,7 +135,7 @@ func checkNvidiaInfoComponent() error {
 
 	for _, ss := range states {
 		for _, s := range ss.States {
-			fmt.Printf("state: %q, healthy: %v, extra info: %q\n", s.Name, s.Healthy, s.ExtraInfo)
+			log.Logger.Infof("state: %q, healthy: %v, extra info: %q\n", s.Name, s.Healthy, s.ExtraInfo)
 		}
 	}